diff --git a/api/models/enums.py b/api/models/enums.py
index c055a45bb4..4434c3fec8 100644
--- a/api/models/enums.py
+++ b/api/models/enums.py
@@ -18,6 +18,6 @@ class WorkflowRunTriggeredFrom(StrEnum):
 
 class DraftVariableType(StrEnum):
     # node means that the correspond variable
-    node = "node"
-    sys = "sys"
-    conversation = "conversation"
+    NODE = "node"
+    SYS = "sys"
+    CONVERSATION = "conversation"
diff --git a/api/models/workflow.py b/api/models/workflow.py
index c01bf1ad06..12e0f347f1 100644
--- a/api/models/workflow.py
+++ b/api/models/workflow.py
@@ -918,19 +918,19 @@ class WorkflowDraftVariable(Base):
         self.value_type = value.value_type
 
     def get_node_id(self) -> str | None:
-        if self.get_variable_type() == DraftVariableType.node:
+        if self.get_variable_type() == DraftVariableType.NODE:
             return self.node_id
         else:
             return None
 
     def get_variable_type(self) -> DraftVariableType:
         match self.node_id:
-            case DraftVariableType.conversation:
-                return DraftVariableType.conversation
-            case DraftVariableType.sys:
-                return DraftVariableType.sys
+            case DraftVariableType.CONVERSATION:
+                return DraftVariableType.CONVERSATION
+            case DraftVariableType.SYS:
+                return DraftVariableType.SYS
             case _:
-                return DraftVariableType.node
+                return DraftVariableType.NODE
 
     @classmethod
     def _create(