diff --git a/litellm/tests/test_key_generate_prisma.py b/litellm/tests/test_key_generate_prisma.py
index e3ad40889..f50df4570 100644
--- a/litellm/tests/test_key_generate_prisma.py
+++ b/litellm/tests/test_key_generate_prisma.py
@@ -658,6 +658,7 @@ def test_key_generate_with_custom_auth(prisma_client):
 
         async def test():
             try:
+                await litellm.proxy.proxy_server.prisma_client.connect()
                 request = GenerateKeyRequest()
                 key = await generate_key_fn(request)
                 pytest.fail(f"Expected an exception. Got {key}")