diff --git a/web/scripts/widgets.js b/web/scripts/widgets.js
index ff5018b7..c128caa5 100644
--- a/web/scripts/widgets.js
+++ b/web/scripts/widgets.js
@@ -279,7 +279,7 @@ export const ComfyWidgets = {
 			),
 		};
 	},
-	TOGGLE(node, inputName, inputData) {
+	BOOLEAN(node, inputName, inputData) {
 		let defaultVal = inputData[1]["default"];
 		return {
 			widget: node.addWidget(
@@ -287,6 +287,7 @@ export const ComfyWidgets = {
 				inputName,
 				defaultVal,
 				() => {},
+				{"on": inputData[1].label_on, "off": inputData[1].label_off}
 				)
 		};
 	},