diff --git a/WAS_Node_Suite.py b/WAS_Node_Suite.py
index 1afb4e4..305d21d 100644
--- a/WAS_Node_Suite.py
+++ b/WAS_Node_Suite.py
@@ -11735,7 +11735,7 @@ def INPUT_TYPES(cls):
             "required": {
                 "image_a": ("IMAGE",),
                 "image_b": ("IMAGE",),
-                "boolean_number": ("NUMBER",),
+                "boolean_number": ("INT", {"default":1, "min":0, "max":1, "step":1}),
             }
         }