diff --git a/comfy_extras/nodes_flux.py b/comfy_extras/nodes_flux.py
index adfd36f71..539635bb2 100644
--- a/comfy_extras/nodes_flux.py
+++ b/comfy_extras/nodes_flux.py
@@ -87,7 +87,7 @@ class ReduxApplyWithAttnMask:
     @classmethod
     def INPUT_TYPES(s):
         return {"required": {
-            "model": ("CONDITIONING", ),
+            "model": ("MODEL", ),
             "conditioning": ("CONDITIONING", ),
             "style_model": ("STYLE_MODEL", ),
             "clip_vision_output": ("CLIP_VISION_OUTPUT", ),