diff --git a/comfy_extras/ui_decorator.py b/comfy_extras/ui_decorator.py
index c5bd339a1..81bb98562 100644
--- a/comfy_extras/ui_decorator.py
+++ b/comfy_extras/ui_decorator.py
@@ -27,7 +27,7 @@ def ui_signal(signals:str|list[str]):
             returns_tuple  = returns_tuple[:-len(signals)]
 
             for i,key in enumerate(signals):
-                returns['ui']['key'] = popped_returns[i]
+                returns_ui['key'] = popped_returns[i]
 
             return { "ui":returns_ui, "result": returns_tuple }
         clazz._ui_signal_decorated_function = _ui_signal_decorated_function