diff --git a/comfy_extras/ui_decorator.py b/comfy_extras/ui_decorator.py
index 81bb98562..a9090ebca 100644
--- a/comfy_extras/ui_decorator.py
+++ b/comfy_extras/ui_decorator.py
@@ -35,6 +35,7 @@ def ui_signal(signals:str|list[str]):
         clazz.OUTPUT_NODE = True
         clazz.UI_OUTPUT = clazz.UI_OUTPUT+"," if hasattr(clazz, 'UI_OUTPUT') else ""
         clazz.UI_OUTPUT += ",".join(signals)
+        return clazz
 
     return decorator