diff --git a/tests/test_litellm/proxy/_experimental/mcp_server/test_mcp_max_concurrent_requests.py b/tests/test_litellm/proxy/_experimental/mcp_server/test_mcp_max_concurrent_requests.py
index e11897b65c2..61d0a4319f2 100644
--- a/tests/test_litellm/proxy/_experimental/mcp_server/test_mcp_max_concurrent_requests.py
+++ b/tests/test_litellm/proxy/_experimental/mcp_server/test_mcp_max_concurrent_requests.py
@@ -46,7 +46,7 @@ def _make_server(server_id: str, max_concurrent_requests: Optional[int]) -> MCPS
 def _patch_client_with_tracker(manager: MCPServerManager, tracker: _ConcurrencyTracker):
     async def fake_create_mcp_client(server, **kwargs):
         class _ProbeClient:
-            async def call_tool(self, params, host_progress_callback=None):
+            async def call_tool(self, params, host_progress_callback=None, persistent_session=None):
                 tracker.enter(server.server_id)
                 try:
                     await asyncio.sleep(HOLD_SECONDS)
diff --git a/tests/test_litellm/proxy/_experimental/mcp_server/test_mcp_server_manager.py b/tests/test_litellm/proxy/_experimental/mcp_server/test_mcp_server_manager.py
index 84595f8d917..1cd39c65c67 100644
--- a/tests/test_litellm/proxy/_experimental/mcp_server/test_mcp_server_manager.py
+++ b/tests/test_litellm/proxy/_experimental/mcp_server/test_mcp_server_manager.py
@@ -6731,7 +6731,7 @@ class TestMCPServerManager:
         # Create mock client that tracks call_tool usage
         mock_client = AsyncMock()
 
-        async def mock_call_tool(params, host_progress_callback=None):
+        async def mock_call_tool(params, host_progress_callback=None, persistent_session=None):
             # Return a mock CallToolResult
             result = MagicMock(spec=CallToolResult)
             result.content = [{"type": "text", "text": "Tool executed successfully"}]