diff --git a/__pycache__/gcn.cpython-37.pyc b/__pycache__/gcn.cpython-37.pyc
deleted file mode 100644
index ea1a271..0000000
Binary files a/__pycache__/gcn.cpython-37.pyc and /dev/null differ
diff --git a/__pycache__/inits.cpython-37.pyc b/__pycache__/inits.cpython-37.pyc
deleted file mode 100644
index 9c44463..0000000
Binary files a/__pycache__/inits.cpython-37.pyc and /dev/null differ
diff --git a/__pycache__/model_sparse.cpython-37.pyc b/__pycache__/model_sparse.cpython-37.pyc
deleted file mode 100644
index 8444d79..0000000
Binary files a/__pycache__/model_sparse.cpython-37.pyc and /dev/null differ
diff --git a/model_sparse.py b/model_sparse.py
index 1f8335f..04e51f9 100644
--- a/model_sparse.py
+++ b/model_sparse.py
@@ -9,7 +9,7 @@
 from gcn import GCNConv
 from torch_scatter import scatter_add
 import torch_sparse
-import torch_sparse_old
+# import torch_sparse_old  # no need now
 from torch_geometric.utils.num_nodes import maybe_num_nodes
 from torch_geometric.utils import remove_self_loops, add_self_loops
 
@@ -111,7 +111,8 @@ def forward(self, A, H_=None):
             a_edge, a_value = result_A[i]
             b_edge, b_value = result_B[i]
             
-            edges, values = torch_sparse_old.spspmm(a_edge, a_value, b_edge, b_value, self.num_nodes, self.num_nodes, self.num_nodes)
+            # edges, values = torch_sparse_old.spspmm(a_edge, a_value, b_edge, b_value, self.num_nodes, self.num_nodes, self.num_nodes)
+            edges, values = torch_sparse.spspmm(a_edge, a_value, b_edge, b_value, self.num_nodes, self.num_nodes, self.num_nodes)
             H.append((edges, values))
         return H, W