diff --git a/.gitlab-ci.yml b/.gitlab-ci.yml
index 7b2ea9b..ef34089 100644
--- a/.gitlab-ci.yml
+++ b/.gitlab-ci.yml
@@ -23,6 +23,7 @@ test:
   script:
     - source $VIRTUAL_ENV/bin/activate
     - pip install -r requirements.txt
+    - export PYTHONPATH=$PYTHONPATH:/home/gitlab-runner/builds/t3_38ooWt/0/mx/heidi-v2/backend
     - pytest --cov=app --cov-report=xml  # Run tests and generate coverage report
 
 lint:
diff --git a/backend/tests/test_auth.py b/backend/tests/test_auth.py
index d80db6a..8104057 100644
--- a/backend/tests/test_auth.py
+++ b/backend/tests/test_auth.py
@@ -1,7 +1,7 @@
 # tests/test_auth.py
 
 from fastapi.testclient import TestClient
-from main import app
+from ..main import app
 
 client = TestClient(app)