From fd6b1af532e22da6764cef34d4eb396fa5bc26c8 Mon Sep 17 00:00:00 2001
From: Faton Ramadani <faton.ramadani14@gmail.com>
Date: Tue, 13 Aug 2024 15:04:47 +0200
Subject: [PATCH] feat(frontend): fix No branch for branch one

---
 frontend/src/lib/components/graph/graphBuilder.ts              | 2 +-
 .../lib/components/graph/renderers/nodes/NoBranchNode.svelte   | 3 ++-
 2 files changed, 3 insertions(+), 2 deletions(-)

diff --git a/frontend/src/lib/components/graph/graphBuilder.ts b/frontend/src/lib/components/graph/graphBuilder.ts
index 5644f99314..5819be0a80 100644
--- a/frontend/src/lib/components/graph/graphBuilder.ts
+++ b/frontend/src/lib/components/graph/graphBuilder.ts
@@ -300,7 +300,7 @@ export default function graphBuilder(
 							...extra
 						},
 						position: { x: -1, y: -1 },
-						type: 'branchOneStart'
+						type: 'noBranch'
 					}
 
 					nodes.push(defaultBranch)
diff --git a/frontend/src/lib/components/graph/renderers/nodes/NoBranchNode.svelte b/frontend/src/lib/components/graph/renderers/nodes/NoBranchNode.svelte
index f58e9557ed..4be680bfff 100644
--- a/frontend/src/lib/components/graph/renderers/nodes/NoBranchNode.svelte
+++ b/frontend/src/lib/components/graph/renderers/nodes/NoBranchNode.svelte
@@ -13,12 +13,13 @@
 		eventHandlers: GraphEventHandlers
 		flowModuleStates: Record<string, GraphModuleState> | undefined
 		offset: number
+		label: string | undefined
 	}
 </script>
 
 <NodeWrapper offset={data.offset} let:darkMode enableSourceHandle enableTargetHandle>
 	<VirtualItem
-		label={'No branches'}
+		label={data.label ?? 'No branches'}
 		modules={data.modules}
 		id={data.id}
 		hideId={true}