diff --git a/src/backend/langflow/api/v1/chat.py b/src/backend/langflow/api/v1/chat.py
index 1cccc654c..4e7807ab8 100644
--- a/src/backend/langflow/api/v1/chat.py
+++ b/src/backend/langflow/api/v1/chat.py
@@ -45,7 +45,7 @@ async def chat(
                 code=status.WS_1008_POLICY_VIOLATION, reason="Unauthorized"
             )
 
-        if client_id in chat_manager.in_memory_cache:
+        if client_id in chat_manager.cache_manager:
             await chat_manager.handle_websocket(client_id, websocket)
         else:
             # We accept the connection but close it immediately