diff --git a/api/core/workflow/nodes/code/code_node.py b/api/core/workflow/nodes/code/code_node.py
index 5dfe398711..0b46f86e9d 100644
--- a/api/core/workflow/nodes/code/code_node.py
+++ b/api/core/workflow/nodes/code/code_node.py
@@ -53,12 +53,12 @@ class CodeNode(BaseNode):
                     ],
                     "code_language": "javascript",
                     "code": JAVASCRIPT_DEFAULT_CODE,
-                    "outputs": [
-                        {
-                            "variable": "result",
-                            "variable_type": "number"
+                    "outputs": {
+                        "result": {
+                            "type": "number",
+                            "children": None
                         }
-                    ]
+                    }
                 }
             }
 
@@ -77,12 +77,12 @@ class CodeNode(BaseNode):
                 ],
                 "code_language": "python3",
                 "code": PYTHON_DEFAULT_CODE,
-                "outputs": [
-                    {
-                        "variable": "result",
-                        "variable_type": "number"
+                "outputs": {
+                    "result": {
+                        "type": "number",
+                        "children": None
                     }
-                ]
+                }
             }
         }