diff --git a/api/core/workflow/nodes/code/entities.py b/api/core/workflow/nodes/code/entities.py
index 0e2b3c99bf..ec3e3fe530 100644
--- a/api/core/workflow/nodes/code/entities.py
+++ b/api/core/workflow/nodes/code/entities.py
@@ -12,7 +12,7 @@ class CodeNodeData(BaseNodeData):
     """
     class Output(BaseModel):
         type: Literal['string', 'number', 'object', 'array[string]', 'array[number]']
-        children: Optional[dict[str, 'CodeNodeData.Output']]
+        children: Optional[dict[str, 'Output']]
 
     variables: list[VariableSelector]
     answer: str