diff --git a/comfy_extras/nodes_autoregressive.py b/comfy_extras/nodes_autoregressive.py
index 6dfcaf766..43b0bfe59 100644
--- a/comfy_extras/nodes_autoregressive.py
+++ b/comfy_extras/nodes_autoregressive.py
@@ -35,7 +35,7 @@ class DecodeTokens:
         return {
             "required": {
                 "clip": (IO.CLIP, {"tooltip": "The model used for generation."}),
-                "tokens": ("GENERATED_TOKENS", ),}
+                "generated_tokens": ("GENERATED_TOKENS", ),}
         }
 
     FUNCTION = "decode"