diff --git a/apps/application/flow/tools.py b/apps/application/flow/tools.py
index 605126a6305..da1678c6bf3 100644
--- a/apps/application/flow/tools.py
+++ b/apps/application/flow/tools.py
@@ -437,7 +437,8 @@ async def anext_async(agen):
 target_source_node_mapping = {
     'TOOL': {'tool-lib-node': lambda n: [n.get('properties').get('node_data').get('tool_lib_id')],
              'ai-chat-node': lambda n: [*(n.get('properties').get('node_data').get('mcp_tool_ids') or []),
-                                        *(n.get('properties').get('node_data').get('tool_ids') or [])]
+                                        *(n.get('properties').get('node_data').get('tool_ids') or [])],
+             'mcp-node': lambda n: [n.get('properties').get('node_data').get('mcp_tool_id')]
              },
     'MODEL': {'ai-chat-node': lambda n: [n.get('properties').get('node_data').get('model_id')],
               'question-node': lambda n: [n.get('properties').get('node_data').get('model_id')],