From 356b8af65d46b024bd037f92b847a544b59af0a0 Mon Sep 17 00:00:00 2001
From: mateo-berri <277851410+mateo-berri@users.noreply.github.com>
Date: Tue, 5 May 2026 16:45:12 -0700
Subject: [PATCH] style: replace with new alias

---
 .../_experimental/mcp_server/test_mcp_stale_session.py    | 8 ++++----
 1 file changed, 4 insertions(+), 4 deletions(-)

diff --git a/tests/test_litellm/proxy/_experimental/mcp_server/test_mcp_stale_session.py b/tests/test_litellm/proxy/_experimental/mcp_server/test_mcp_stale_session.py
index 078c5414e59..96be37bbd42 100644
--- a/tests/test_litellm/proxy/_experimental/mcp_server/test_mcp_stale_session.py
+++ b/tests/test_litellm/proxy/_experimental/mcp_server/test_mcp_stale_session.py
@@ -496,7 +496,7 @@ async def test_per_user_oauth_missing_stored_token_returns_preemptive_401():
     try:
         from litellm.proxy._experimental.mcp_server.server import (
             handle_streamable_http_mcp,
-            session_manager,
+            session_manager_stateless,
         )
     except ImportError:
         pytest.skip("MCP server not available")
@@ -545,7 +545,7 @@ async def test_per_user_oauth_missing_stored_token_returns_preemptive_401():
             return_value=oauth_server,
         ),
         patch.object(
-            session_manager,
+            session_manager_stateless,
             "handle_request",
             new_callable=AsyncMock,
         ) as mock_handle_request,
@@ -569,7 +569,7 @@ async def test_per_user_oauth_with_stored_token_skips_preemptive_401():
     try:
         from litellm.proxy._experimental.mcp_server.server import (
             handle_streamable_http_mcp,
-            session_manager,
+            session_manager_stateless,
         )
     except ImportError:
         pytest.skip("MCP server not available")
@@ -618,7 +618,7 @@ async def test_per_user_oauth_with_stored_token_skips_preemptive_401():
             return_value=oauth_server,
         ),
         patch.object(
-            session_manager,
+            session_manager_stateless,
             "handle_request",
             new_callable=AsyncMock,
         ) as mock_handle_request,