diff --git a/bin/train.py b/bin/train.py
index 7938b24..866ffc2 100644
--- a/bin/train.py
+++ b/bin/train.py
@@ -27,7 +27,8 @@
 from pytorch_lightning.strategies.ddp import DDPStrategy
 
 from transformers import BertConfig
-
+# Add parent directory to Python path
+sys.path.append(os.path.abspath(os.path.join(os.path.dirname(__file__), "..")))
 from foldingdiff import datasets
 from foldingdiff import modelling
 from foldingdiff import losses