diff --git a/bin/sample.py b/bin/sample.py
index 97cfccf..f3957b5 100644
--- a/bin/sample.py
+++ b/bin/sample.py
@@ -22,7 +22,7 @@
 # Import data loading code from main training script
 from train import get_train_valid_test_sets
 from annot_secondary_structures import make_ss_cooccurrence_plot
-
+sys.path.append(os.path.abspath(os.path.join(os.path.dirname(__file__), "..")))
 from foldingdiff import modelling
 from foldingdiff import sampling
 from foldingdiff import plotting