diff --git a/tests/test_train.py b/tests/test_train.py
index edf0cbf..55c4175 100644
--- a/tests/test_train.py
+++ b/tests/test_train.py
@@ -1,4 +1,13 @@
 import pytest
+import os
+import sys
+
+# Get the parent directory
+parent_dir = os.path.dirname(os.path.realpath(__file__))
+
+# Add the parent directory to sys.path
+sys.path.append(parent_dir)
+
 from src.models.predict_model import predict,SentiBites
 from src.models.train_model import pre_processing, train
 import datasets