diff --git a/api/core/workflow/nodes/base/node.py b/api/core/workflow/nodes/base/node.py
index ae807ec7a1..3a5e6c1269 100644
--- a/api/core/workflow/nodes/base/node.py
+++ b/api/core/workflow/nodes/base/node.py
@@ -206,7 +206,7 @@ class Node(Generic[NodeDataT]):
     def __init__(
         self,
         id: str,
-        config: dict[str, Any],  # NodeConfigDict
+        config: dict[str, Any] | NodeConfigDict,  # NodeConfigDict
         graph_init_params: GraphInitParams,
         graph_runtime_state: GraphRuntimeState,
     ) -> None: