From 0843af2996df7ac877029fe6db9bdf1cdb3bfbfc Mon Sep 17 00:00:00 2001
From: Joel <iamjoel007@gmail.com>
Date: Mon, 1 Apr 2024 12:39:54 +0800
Subject: [PATCH] fix: sys var not show

---
 web/app/components/workflow/hooks/use-workflow.ts | 5 +++--
 1 file changed, 3 insertions(+), 2 deletions(-)

diff --git a/web/app/components/workflow/hooks/use-workflow.ts b/web/app/components/workflow/hooks/use-workflow.ts
index f179a441b3..9d8d3d4190 100644
--- a/web/app/components/workflow/hooks/use-workflow.ts
+++ b/web/app/components/workflow/hooks/use-workflow.ts
@@ -342,10 +342,11 @@ export const useWorkflow = () => {
       setViewport(viewport)
   }, [store, reactflow])
 
-  const getNode = useCallback((nodeId: string) => {
+  const getNode = useCallback((nodeId?: string) => {
     const { getNodes } = store.getState()
+    const nodes = getNodes()
 
-    return getNodes().find(node => node.id === nodeId)
+    return nodes.find(node => node.id === nodeId) || nodes.find(node => node.data.type === BlockEnum.Start)
   }, [store])
 
   return {