diff --git a/tests/test_litellm/proxy/_experimental/mcp_server/test_mcp_server.py b/tests/test_litellm/proxy/_experimental/mcp_server/test_mcp_server.py index 0513650e1ff..d630ba6aaa5 100644 --- a/tests/test_litellm/proxy/_experimental/mcp_server/test_mcp_server.py +++ b/tests/test_litellm/proxy/_experimental/mcp_server/test_mcp_server.py @@ -967,10 +967,9 @@ async def test_oauth2_headers_passed_to_mcp_client(): global_mcp_server_manager, "_fetch_tools_with_timeout", side_effect=mock_fetch_tools_with_timeout, - ), patch.object( - global_mcp_server_manager, - "get_allowed_mcp_servers", - AsyncMock(return_value=[oauth2_server.server_id]), + ), patch( + "litellm.proxy._experimental.mcp_server.server._get_allowed_mcp_servers", + AsyncMock(return_value=[oauth2_server]), ): # Call _get_tools_from_mcp_servers which should eventually call _create_mcp_client await _get_tools_from_mcp_servers(