diff --git a/web/scripts/app.js b/web/scripts/app.js
index 695f634ab..099fc5738 100644
--- a/web/scripts/app.js
+++ b/web/scripts/app.js
@@ -551,10 +551,10 @@ class ComfyApp {
 			const nodeData = defs[nodeId];
 			const node = Object.assign(
 				function ComfyNode() {
-					const inputs = nodeData["input"]["required"];
+					const inputs = Object.assign(nodeData["input"]["required"], nodeData["input"]["widget"]);
 					const config = { minWidth: 1, minHeight: 1 };
 					for (const inputName in inputs) {
-						const inputs = Object.assign(nodeData["input"]["required"], nodeData["input"]["widget"]);
+						const inputData = inputs[inputName];
 						const type = inputData[0];
 
 						if (Array.isArray(type)) {
diff --git a/web/scripts/widgets.js b/web/scripts/widgets.js
index 56e9c533f..3ff059337 100644
--- a/web/scripts/widgets.js
+++ b/web/scripts/widgets.js
@@ -51,6 +51,7 @@ function imagesendWidget(node, inputName, inputData, app) {
 
 			const image_name = node.images[0].filename;
 			const copied = false;
+			const imageWidget = node.widgets.find((w) => w.name === "image");
 
 			for(let i in app.graph._nodes) {
 					var n = app.graph._nodes[i];
@@ -64,6 +65,7 @@ function imagesendWidget(node, inputName, inputData, app) {
 											await api.sendOutputToInputImage(image_name);
 									}
 
+									imageWidget.value = data.name;
 									const thatImageWidget = n.widgets.find((w) => w.value === "image");
 									await showImage(n,thatImageWidget,image_name);
 							}