diff --git a/comfy-nodes/input_text.py b/comfy-nodes/input_text.py
index fb90d7a..da19009 100755
--- a/comfy-nodes/input_text.py
+++ b/comfy-nodes/input_text.py
@@ -150,8 +150,8 @@ class ShellAgentPluginInputInteger:
             }
         }
 
-    RETURN_TYPES = ("FLOAT",)
-    RETURN_NAMES = ("float",)
+    RETURN_TYPES = ("INT",)
+    RETURN_NAMES = ("int",)
 
     FUNCTION = "run"