diff --git a/apps/api/src/database.py b/apps/api/src/database.py
index 2fc1434..adbf470 100644
--- a/apps/api/src/database.py
+++ b/apps/api/src/database.py
@@ -44,8 +44,11 @@ async def init_database(
         try:
             # Test basic connectivity
             from sqlalchemy import text
-            async with engine.connect() as conn:
+            conn = await engine.connect()
+            try:
                 await conn.execute(text("SELECT 1"))
+            finally:
+                await conn.close()
 
             logger.info("Database connection established.")