diff --git a/train_code/selective_search/core.py b/train_code/selective_search/core.py
index 502f9e9..63609dc 100755
--- a/train_code/selective_search/core.py
+++ b/train_code/selective_search/core.py
@@ -1,7 +1,7 @@
 from joblib import Parallel, delayed
 from skimage.segmentation import felzenszwalb
 from util import oversegmentation, switch_color_space, load_strategy
-from structure import HierarchicalGrouping
+from selective_search.structure import HierarchicalGrouping