diff --git a/comfy_extras/ui_decorator.py b/comfy_extras/ui_decorator.py
index 03f84506a..17de0d366 100644
--- a/comfy_extras/ui_decorator.py
+++ b/comfy_extras/ui_decorator.py
@@ -27,7 +27,8 @@ 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]
+                if popped_returns[i] is not None:
+                    returns_ui[key] = popped_returns[i]
 
             return { "ui":returns_ui, "result": returns_tuple }
         clazz._ui_signal_decorated_function = _ui_signal_decorated_function