diff --git a/frontend/src/lib/components/ArgInput.svelte b/frontend/src/lib/components/ArgInput.svelte
index 4dc1184eca..1a634b42e2 100644
--- a/frontend/src/lib/components/ArgInput.svelte
+++ b/frontend/src/lib/components/ArgInput.svelte
@@ -762,6 +762,7 @@
 															{onlyMaskPassword}
 															{disablePortal}
 															{disabled}
+															{prettifyHeader}
 															schema={getSchemaFromProperties(itemsType?.properties)}
 															bind:args={value[i]}
 														/>
@@ -904,6 +905,7 @@
 												{onlyMaskPassword}
 												{disablePortal}
 												{disabled}
+												{prettifyHeader}
 												bind:schema={
 													() => ({
 														properties: obj.properties ?? {},
@@ -934,6 +936,7 @@
 												{onlyMaskPassword}
 												{disablePortal}
 												{disabled}
+												{prettifyHeader}
 												hiddenArgs={['label', 'kind']}
 												schema={{
 													properties: obj.properties,
@@ -1009,6 +1012,7 @@
 							{onlyMaskPassword}
 							{disablePortal}
 							{disabled}
+							{prettifyHeader}
 							bind:schema={
 								() => ({
 									properties,
@@ -1044,6 +1048,7 @@
 							{onlyMaskPassword}
 							{disablePortal}
 							{disabled}
+							{prettifyHeader}
 							schema={{
 								properties,
 								order,
diff --git a/frontend/src/lib/components/FlowPreviewResult.svelte b/frontend/src/lib/components/FlowPreviewResult.svelte
index d92192fadb..ceaebe5b15 100644
--- a/frontend/src/lib/components/FlowPreviewResult.svelte
+++ b/frontend/src/lib/components/FlowPreviewResult.svelte
@@ -1,5 +1,5 @@
 <script lang="ts">
-	import { Badge, Loader2 } from 'lucide-svelte'
+	import { Loader2 } from 'lucide-svelte'
 	import FlowJobResult from './FlowJobResult.svelte'
 	import FlowPreviewStatus from './preview/FlowPreviewStatus.svelte'
 	import FlowStatusWaitingForEvents from './FlowStatusWaitingForEvents.svelte'
@@ -7,6 +7,7 @@
 	import { emptyString } from '$lib/utils'
 	import type { DurationStatus } from './graph'
 	import type { Writable } from 'svelte/store'
+	import Badge from './Badge.svelte'
 
 	interface Props {
 		job: Job
@@ -81,10 +82,12 @@
 			{#if mod.type == 'InProgress'}
 				{@const rawMod = job.raw_flow?.modules[i]}
 
-				<div
-					><span class="inline-flex gap-1"
-						><Badge color="indigo">{mod.id}</Badge>
-						<span class="font-medium text-primary">
+				<div>
+					<span class="inline-flex gap-1">
+						<div class="text-xs flex items-center px-2 py-2">
+							<Badge>{mod.id}</Badge>
+						</div>
+						<span class="font-medium text-primary mt-0.5">
 							{#if !emptyString(rawMod?.summary)}
 								{rawMod?.summary ?? ''}
 							{:else if rawMod?.value.type == 'script'}
@@ -94,7 +97,7 @@
 							{/if}
 						</span>
 
-						<Loader2 class="animate-spin" /></span
+						<Loader2 class="animate-spin mt-0.5" /></span
 					></div
 				>
 			{/if}