diff --git a/image_c_postprocessor.py b/image_c_postprocessor.py
index bc169ec..00280ff 100644
--- a/image_c_postprocessor.py
+++ b/image_c_postprocessor.py
@@ -148,7 +148,7 @@ class PrivateImageMask:
     def INPUT_TYPES(s):
         return {
             "required": {
-                "input_image": ("IMAGE", ),
+                "image": ("IMAGE", ),
                 "mask_image": ("IMAGE", ),
                 "mask_gray_threshold": ("INT", {"default": 127})
             },