diff --git a/frontend/src/lib/components/graph/FlowGraph.svelte b/frontend/src/lib/components/graph/FlowGraph.svelte
index c352402952..45fa9efbdc 100644
--- a/frontend/src/lib/components/graph/FlowGraph.svelte
+++ b/frontend/src/lib/components/graph/FlowGraph.svelte
@@ -522,7 +522,7 @@
 				}
 			} else if (module.value.type == 'branchone') {
 				if (flowModuleStates?.[module.id]?.branchChosen == i) {
-					borderStatus = 'Success'
+					borderStatus = flowModuleStates?.[module.id]?.type
 				}
 			}
 			items.push(