diff --git a/comfy_extras/nodes/nodes_api.py b/comfy_extras/nodes/nodes_api.py
index 1f7a2affd..75c0bd477 100644
--- a/comfy_extras/nodes/nodes_api.py
+++ b/comfy_extras/nodes/nodes_api.py
@@ -16,9 +16,7 @@ class OutputTensor(CustomNode):
     FUNCTION = "execute"
 
     def execute(self, tensor):
-        return {
-            "result": tensor
-        }
+        return {"ui": {"tensor": tensor}}
 
 
 export_custom_nodes()