diff --git a/litellm/tests/test_key_generate_prisma.py b/litellm/tests/test_key_generate_prisma.py
index 27a986b9c..f53e241ce 100644
--- a/litellm/tests/test_key_generate_prisma.py
+++ b/litellm/tests/test_key_generate_prisma.py
@@ -1779,7 +1779,7 @@ async def test_upperbound_key_params(prisma_client):
 
 
 def test_get_bearer_token():
-    from litellm.proxy.proxy_server import _get_bearer_token
+    from litellm.proxy.auth.user_api_key_auth import _get_bearer_token
 
     # Test valid Bearer token
     api_key = "Bearer valid_token"