diff --git a/api.py b/api.py index 4fa2e226f..484dc574a 100644 --- a/api.py +++ b/api.py @@ -208,7 +208,7 @@ async def drop_db(payload: Payload): else: pass - from cognitive_architecture.database.postgres import drop_database, create_admin_engine + from cognitive_architecture.database.create_database import drop_database, create_admin_engine engine = create_admin_engine(username, password, host, database_name) drop_database(engine)