diff --git a/apps/application/flow/step_node/mcp_node/impl/base_mcp_node.py b/apps/application/flow/step_node/mcp_node/impl/base_mcp_node.py
index b08a7e02a..6dd480485 100644
--- a/apps/application/flow/step_node/mcp_node/impl/base_mcp_node.py
+++ b/apps/application/flow/step_node/mcp_node/impl/base_mcp_node.py
@@ -4,7 +4,6 @@ import json
 from typing import List
 
 from langchain_mcp_adapters.client import MultiServerMCPClient
-from langchain_mcp_adapters.sessions import create_session
 
 from application.flow.i_step_node import NodeResult
 from application.flow.step_node.mcp_node.i_mcp_node import IMcpNode
@@ -25,7 +24,7 @@ class BaseMcpNode(IMcpNode):
 
         async def call_tool(t, a):
             client = MultiServerMCPClient(servers)
-            async with create_session(client.connections[mcp_server]) as s:
+            async with client.session(mcp_server) as s:
                 return await s.call_tool(t, a)
 
         res = asyncio.run(call_tool(mcp_tool, params))