diff --git a/cognee/modules/tasks/create_task_status_table.py b/cognee/modules/tasks/create_task_status_table.py
index 26beceb1c..5fa99e9bf 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.InfrastructureConfig import infrastructure_config
-
+from cognee.infrastructure.databases.relational.config import get_relationaldb_config
+config = get_relationaldb_config()
 def create_task_status_table():
-    db_engine = infrastructure_config.get_config("database_engine")
+    db_engine = config.database_engine
 
     db_engine.create_table("cognee_task_status", [
         dict(name = "data_id", type = "STRING"),