diff --git a/cognee/modules/tasks/create_task_status_table.py b/cognee/modules/tasks/create_task_status_table.py
index 5fa99e9bf..ec9e045aa 100644
--- a/cognee/modules/tasks/create_task_status_table.py
+++ b/cognee/modules/tasks/create_task_status_table.py
@@ -1,7 +1,7 @@
 from cognee.infrastructure.databases.relational.config import get_relationaldb_config
 config = get_relationaldb_config()
 def create_task_status_table():
-    db_engine = config.database_engine
+    db_engine = config.db_engine
 
     db_engine.create_table("cognee_task_status", [
         dict(name = "data_id", type = "STRING"),