diff --git a/README.md b/README.md
index 5be845c16..007d4a1b7 100644
--- a/README.md
+++ b/README.md
@@ -1,5 +1,4 @@
 # PromethAI-Memory
-
 Memory management and testing for the AI Applications and RAGs
 
 
@@ -148,4 +147,3 @@ After that, you can run the RAG test manager.
 ```
 
 Examples of metadata structure and test set are in the folder "example_data"
-
diff --git a/level_3/scripts/create_database.py b/level_3/scripts/create_database.py
index 05681a6b7..b06021ddd 100644
--- a/level_3/scripts/create_database.py
+++ b/level_3/scripts/create_database.py
@@ -47,19 +47,20 @@ def create_database(username, password, host, db_name):
     engine.dispose()
 
 
-def create_tables():
+def create_tables(engine):
     Base.metadata.create_all(bind=engine)
 
-
 if __name__ == "__main__":
     username = os.getenv('POSTGRES_USER')
     password = os.getenv('POSTGRES_PASSWORD')
     database_name = os.getenv('POSTGRES_DB')
     host = os.getenv('POSTGRES_HOST')
 
+    engine = create_admin_engine(username, password, host, database_name)
+
     if not database_exists(username, password, host, database_name):
         print(f"Database {database_name} does not exist. Creating...")
         create_database(username, password, host, database_name)
         print(f"Database {database_name} created successfully.")
 
-    create_tables()
\ No newline at end of file
+    create_tables(engine)
\ No newline at end of file