diff --git a/web/scripts/ui.js b/web/scripts/ui.js
index 8134e341..cb31f10e 100644
--- a/web/scripts/ui.js
+++ b/web/scripts/ui.js
@@ -258,7 +258,12 @@ export class ComfyUI {
 			$el("button.comfy-queue-btn", { textContent: "Queue Prompt", onclick: () => app.queuePrompt(0, this.batchCount) }),
 			$el("div", {}, [
 				$el("label", { innerHTML: "Extra options"}, [
-					$el("input", { type: "checkbox", onchange: (i) => document.getElementById('extraOptions').style.visibility = i.srcElement.checked ? "visible" : "collapse" })
+					$el("input", { type: "checkbox", 
+						onchange: (i) => { 
+							document.getElementById('extraOptions').style.visibility = i.srcElement.checked ? "visible" : "collapse";
+							this.batchCount = i.srcElement.checked ? document.getElementById('batchCountInputRange').value : 1;
+						}
+					})
 				])
 			]),
 			$el("div", { id: "extraOptions", style: { width: "100%", visibility: "collapse" }}, [