From ff92e2f8979e466b5204f96eab6ebc0696275c23 Mon Sep 17 00:00:00 2001
From: Lucas Abel <22837557+uael@users.noreply.github.com>
Date: Wed, 27 Nov 2024 16:07:16 +0100
Subject: [PATCH] fix simple step flow rendering for while loop (#4804)

---
 frontend/src/lib/components/FlowStatusViewerInner.svelte | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/frontend/src/lib/components/FlowStatusViewerInner.svelte b/frontend/src/lib/components/FlowStatusViewerInner.svelte
index 8878fcfeed..c6de80ac14 100644
--- a/frontend/src/lib/components/FlowStatusViewerInner.svelte
+++ b/frontend/src/lib/components/FlowStatusViewerInner.svelte
@@ -713,7 +713,7 @@
 
 			if (jobLoaded.job_kind == 'script' || jobLoaded.job_kind == 'flowscript' || jobLoaded.job_kind == 'preview') {
 				let id: string | undefined = undefined
-				if (innerModule?.type == 'forloopflow' && innerModule.modules.length == 1) {
+				if ((innerModule?.type == 'forloopflow' || innerModule?.type == 'whileloopflow') && innerModule.modules.length == 1) {
 					id = innerModule?.modules?.[0]?.id
 				}
 				if (id) {