diff --git a/apps/application/flow/common.py b/apps/application/flow/common.py
index 98f706fde..2ba944875 100644
--- a/apps/application/flow/common.py
+++ b/apps/application/flow/common.py
@@ -20,7 +20,8 @@ from models_provider.tools import get_model_credential
 from tools.models.tool import Tool
 
 end_nodes = ['ai-chat-node', 'reply-node', 'function-node', 'function-lib-node', 'application-node',
-             'image-understand-node', 'speech-to-text-node', 'text-to-speech-node', 'image-generate-node']
+             'image-understand-node', 'speech-to-text-node', 'text-to-speech-node', 'image-generate-node',
+             'variable-assign-node']
 
 
 class Answer:
diff --git a/ui/src/workflow/common/validate.ts b/ui/src/workflow/common/validate.ts
index 3d781af9e..50e7e94e9 100644
--- a/ui/src/workflow/common/validate.ts
+++ b/ui/src/workflow/common/validate.ts
@@ -1,6 +1,6 @@
-import {WorkflowType, WorkflowMode} from '@/enums/application'
+import { WorkflowType, WorkflowMode } from '@/enums/application'
 
-import {t} from '@/locales'
+import { t } from '@/locales'
 
 const end_nodes: Array<string> = [
   WorkflowType.AiChat,
@@ -18,7 +18,8 @@ const end_nodes: Array<string> = [
   WorkflowType.LoopBodyNode,
   WorkflowType.LoopNode,
   WorkflowType.LoopBreakNode,
-  WorkflowType.VideoUnderstandNode
+  WorkflowType.VideoUnderstandNode,
+  WorkflowType.VariableAssignNode,
 ]
 
 const loop_end_nodes: Array<string> = [