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) {