diff --git a/graphxai/utils/perturb/perturb.py b/graphxai/utils/perturb/perturb.py
index 1d3fba61..c8232834 100644
--- a/graphxai/utils/perturb/perturb.py
+++ b/graphxai/utils/perturb/perturb.py
@@ -3,7 +3,7 @@
 from typing import List
 from torch_geometric.data import Data
 from torch_geometric.utils import convert, to_undirected
-from torch_geometric.utils.subgraph import k_hop_subgraph
+from torch_geometric.utils import k_hop_subgraph
 from networkx.linalg.graphmatrix import adjacency_matrix as adj_mat
 
 from .nx_modified import swap