diff --git a/packages/editor-ui/src/views/NodeView.vue b/packages/editor-ui/src/views/NodeView.vue
index 14ec069210..1292bd59dc 100644
--- a/packages/editor-ui/src/views/NodeView.vue
+++ b/packages/editor-ui/src/views/NodeView.vue
@@ -2743,7 +2743,11 @@ export default defineComponent({
 
 								if (nodeType && inputs.length === 1) {
 									this.pullConnActiveNodeName = node.name;
-									const endpointUUID = this.getInputEndpointUUID(nodeName, 0);
+									const endpointUUID = this.getInputEndpointUUID(
+										nodeName,
+										connection.parameters.type,
+										0,
+									);
 									if (endpointUUID) {
 										const endpoint = this.instance?.getEndpoint(endpointUUID);