diff --git a/frontend/src/lib/components/flows/content/FlowModule.svelte b/frontend/src/lib/components/flows/content/FlowModule.svelte
index 789eda5ecb..720966f945 100644
--- a/frontend/src/lib/components/flows/content/FlowModule.svelte
+++ b/frontend/src/lib/components/flows/content/FlowModule.svelte
@@ -109,6 +109,7 @@
 			<FlowModuleHeader
 				bind:module={flowModule}
 				on:delete
+				on:toggleSuspend={() => (selected = 'suspend')}
 				on:toggleRetry={() => (selected = 'retries')}
 				on:toggleStopAfterIf={() => (selected = 'early-stop')}
 				on:fork={() => apply(fork, flowModule)}
diff --git a/frontend/src/lib/components/flows/header/FlowPreviewButtons.svelte b/frontend/src/lib/components/flows/header/FlowPreviewButtons.svelte
index e38a5b3143..11d1468c72 100644
--- a/frontend/src/lib/components/flows/header/FlowPreviewButtons.svelte
+++ b/frontend/src/lib/components/flows/header/FlowPreviewButtons.svelte
@@ -17,7 +17,8 @@
 		'settings-schedule',
 		'settings-retries',
 		'inputs',
-		'schedules'
+		'schedules',
+		'failure'
 	].includes($selectedId)
 </script>
 
diff --git a/frontend/src/lib/components/flows/map/FlowModuleSchemaMap.svelte b/frontend/src/lib/components/flows/map/FlowModuleSchemaMap.svelte
index cb190c7226..8de195a59c 100644
--- a/frontend/src/lib/components/flows/map/FlowModuleSchemaMap.svelte
+++ b/frontend/src/lib/components/flows/map/FlowModuleSchemaMap.svelte
@@ -51,7 +51,7 @@
 				)}
 			>
 				<Icon data={faSliders} class="mr-2" />
-				<span class="font-bold flex flex-row justify-between w-full flex-wrap gap-2"
+				<span class="font-bold flex flex-row justify-between w-full flex-wrap gap-2 items-center"
 					>Settings <span
 						class={classNames('badge', $schedule?.enabled ? 'badge-on' : 'badge-off')}
 						on:click|stopPropagation={() => select('settings-schedule')}