From f076a1aefd920232db911c4748963c6a0fa2d880 Mon Sep 17 00:00:00 2001
From: Chris <chris.goringe@gmail.com>
Date: Mon, 11 Sep 2023 17:30:00 +1000
Subject: [PATCH] return the clazz!

---
 comfy_extras/ui_decorator.py | 1 +
 1 file changed, 1 insertion(+)

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