diff --git a/tests/store_model_in_db_tests/test_mcp_servers.py b/tests/store_model_in_db_tests/test_mcp_servers.py
index 2402b012b4..21b12fe182 100644
--- a/tests/store_model_in_db_tests/test_mcp_servers.py
+++ b/tests/store_model_in_db_tests/test_mcp_servers.py
@@ -9,7 +9,8 @@ import os
 from starlette import status
 
 from litellm.constants import LITELLM_PROXY_ADMIN_NAME
-from litellm.proxy._types import MCPAuth, MCPSpecVersion, MCPSpecVersionType, MCPTransportType, MCPTransport, NewMCPServerRequest, LiteLLM_MCPServerTable
+from litellm.proxy._types import MCPSpecVersion, MCPSpecVersionType, MCPTransportType, MCPTransport, NewMCPServerRequest, LiteLLM_MCPServerTable
+from litellm.types.mcp import MCPAuth
 from litellm.proxy.management_endpoints.mcp_management_endpoints import does_mcp_server_exist
 
 TEST_MASTER_KEY = os.getenv("LITELLM_MASTER_KEY", "sk-1234")