diff --git a/api/core/workflow/nodes/tool/tool_node.py b/api/core/workflow/nodes/tool/tool_node.py
index bdc9cc9da6..79e0073df6 100644
--- a/api/core/workflow/nodes/tool/tool_node.py
+++ b/api/core/workflow/nodes/tool/tool_node.py
@@ -196,6 +196,8 @@ class ToolNode(BaseNode[ToolNodeData]):
                 else:
                     transfer_method = FileTransferMethod.TOOL_FILE
 
+                tool_file_id = str(url).split("/")[-1].split(".")[0]
+
                 with Session(db.engine) as session:
                     stmt = select(ToolFile).where(ToolFile.id == tool_file_id)
                     tool_file = session.scalar(stmt)