diff --git a/comfy_extras/ui_decorator.py b/comfy_extras/ui_decorator.py
index 03f84506a..c5bd339a1 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
@@ -35,7 +35,6 @@ 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
         
diff --git a/web/extensions/core/widgetInputs.js b/web/extensions/core/widgetInputs.js
index 09179a57b..606605f0a 100644
--- a/web/extensions/core/widgetInputs.js
+++ b/web/extensions/core/widgetInputs.js
@@ -158,7 +158,7 @@ app.registerExtension({
 
 			if (this.inputs) {
 				for (const input of this.inputs) {
-					if (input.widget && !input.widget.config[1]?.forceInput && !input.widget.config[1]?.defaultInput) {
+					if (input.widget && !input.widget.config[1]?.forceInput) {
 						const w = this.widgets.find((w) => w.name === input.widget.name);
 						if (w) {
 							hideWidget(this, w);