diff --git a/web/scripts/app.js b/web/scripts/app.js
index 8b273f626..8c9e7a27f 100644
--- a/web/scripts/app.js
+++ b/web/scripts/app.js
@@ -1351,35 +1351,39 @@ export class ComfyApp {
 				let parent = node.getInputNode(i);
 				if (parent) {
 					let link = node.getInputLink(i);
-					while (parent.mode === 4) {
+					while (parent.mode === 4 || parent.isVirtualNode) {
 						let found = false;
-						if (link) {
-							let all_inputs = [link.origin_slot].concat(parent.inputs)
-							for (let parent_input in all_inputs) {
-								if (parent.inputs[parent_input].type === node.inputs[i].type) {
-									link = parent.getInputLink(parent_input);
-									if (link) {
-										parent = parent.getInputNode(parent_input);
-									}
+						if (parent.isVirtualNode) {
+							link = parent.getInputLink(link.origin_slot);
+							if (link) {
+								parent = parent.getInputNode(link.origin_slot);
+								if (parent) {
 									found = true;
-									break;
+								}
+							}
+						} else if (link && parent.mode === 4) {
+							let all_inputs = [link.origin_slot];
+							if (parent.inputs) {
+								all_inputs = all_inputs.concat(Object.keys(parent.inputs))
+								for (let parent_input in all_inputs) {
+									parent_input = all_inputs[parent_input];
+									if (parent.inputs[parent_input].type === node.inputs[i].type) {
+										link = parent.getInputLink(parent_input);
+										if (link) {
+											parent = parent.getInputNode(parent_input);
+										}
+										found = true;
+										break;
+									}
 								}
 							}
 						}
+
 						if (!found) {
 							break;
 						}
 					}
 
-					while (parent && parent.isVirtualNode) {
-						link = parent.getInputLink(link.origin_slot);
-						if (link) {
-							parent = parent.getInputNode(link.origin_slot);
-						} else {
-							parent = null;
-						}
-					}
-
 					if (link) {
 						inputs[node.inputs[i].name] = [String(link.origin_id), parseInt(link.origin_slot)];
 					}