diff --git a/web/scripts/app.js b/web/scripts/app.js
index 861db16b..8598d447 100644
--- a/web/scripts/app.js
+++ b/web/scripts/app.js
@@ -411,7 +411,7 @@ export class ComfyApp {
 		node.prototype.setSizeForImage = function (force) {
 			if(!force && this.animatedImages) return;
 
-			if (this.inputHeight) {
+			if (this.inputHeight || this.freeWidgetSpace > 210) {
 				this.setSize(this.size);
 				return;
 			}
diff --git a/web/scripts/domWidget.js b/web/scripts/domWidget.js
index 37d26f3c..e919428a 100644
--- a/web/scripts/domWidget.js
+++ b/web/scripts/domWidget.js
@@ -120,6 +120,8 @@ function computeSize(size) {
 		freeSpace -= 220;
 	}
 
+	this.freeWidgetSpace = freeSpace;
+
 	if (freeSpace < 0) {
 		// Not enough space for all widgets so we need to grow
 		size[1] -= freeSpace;