diff --git a/tests/test_database.py b/tests/test_database.py
index a2bdd6b..a9529df 100644
--- a/tests/test_database.py
+++ b/tests/test_database.py
@@ -4,7 +4,7 @@ from pathlib import Path
 from sqlalchemy import inspect, Column, Integer, String
 from sqlalchemy.orm import Session
 
-from src.pilgrim.database import Database,Base
+from pilgrim.database import Database,Base
 
 class MockUser(Base):
     __tablename__ = 'mock_users'