diff --git a/tests/test_e2e_postgres_migration.py b/tests/test_e2e_postgres_migration.py
index 24e8b9c5..c6065d78 100644
--- a/tests/test_e2e_postgres_migration.py
+++ b/tests/test_e2e_postgres_migration.py
@@ -31,6 +31,11 @@ def pg_config():
         "database": os.getenv("POSTGRES_DB", "lightrag_test"),
         "workspace": os.getenv("POSTGRES_WORKSPACE", "e2e_test"),
         "max_connections": 10,
+        # Connection retry configuration
+        "connection_retry_attempts": 3,
+        "connection_retry_backoff": 0.5,
+        "connection_retry_backoff_max": 5.0,
+        "pool_close_timeout": 5.0,
     }