diff --git a/tests/test_db.py b/tests/test_db.py
index 5dea240..0843a21 100644
--- a/tests/test_db.py
+++ b/tests/test_db.py
@@ -1,4 +1,4 @@
-from ..db import DB
+from gpt_engineer.db import DB
 
 
 def test_db():