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 {