diff --git a/dgl_ptm/dgl_ptm/agent/agent_update.py b/dgl_ptm/dgl_ptm/agent/agent_update.py
index b71d16b..d4415cd 100644
--- a/dgl_ptm/dgl_ptm/agent/agent_update.py
+++ b/dgl_ptm/dgl_ptm/agent/agent_update.py
@@ -2,7 +2,6 @@
 from dgl_ptm.agent.wealth_consumption import wealth_consumption
 from dgl_ptm.agent.capital_update import capital_update
 from dgl_ptm.util.network_metrics import node_degree, node_weighted_degree
-import torch
 
 
 def agent_update(model_graph, model_params=None, device=None, timestep=None, method='pseudo'):
diff --git a/dgl_ptm/dgl_ptm/model/step.py b/dgl_ptm/dgl_ptm/model/step.py
index 479add3..0e89f80 100644
--- a/dgl_ptm/dgl_ptm/model/step.py
+++ b/dgl_ptm/dgl_ptm/model/step.py
@@ -11,8 +11,6 @@
 from dgl_ptm.network.local_attachment import local_attachment
 from dgl_ptm.network.local_attachment_basic_homophily import local_attachment_homophily
 from dgl_ptm.network.random_edge_noise import random_edge_noise
-from dgl_ptm.util.network_metrics import average_degree, average_weighted_degree
-
 
 def ptm_step(agent_graph, device, timestep, params):
     """Step - time-stepping module for the poverty-trap model.
diff --git a/dgl_ptm/tests/test_util.py b/dgl_ptm/tests/test_util.py
index c80407a..8b28458 100644
--- a/dgl_ptm/tests/test_util.py
+++ b/dgl_ptm/tests/test_util.py
@@ -21,7 +21,7 @@ def model():
 
 class TestNetworkMetrics:
     def test_average_degree_initialize(self, model):
-        expected_ad = 1.98
+        expected_ad = 3.96
         assert model.average_degree == expected_ad
 
     def test_average_degree_step(self, model):