diff --git a/frontend/src/lib/components/flows/content/FlowModuleHeader.svelte b/frontend/src/lib/components/flows/content/FlowModuleHeader.svelte
index 63d053fe14..a1e957a045 100644
--- a/frontend/src/lib/components/flows/content/FlowModuleHeader.svelte
+++ b/frontend/src/lib/components/flows/content/FlowModuleHeader.svelte
@@ -3,7 +3,6 @@
 	import { type FlowModule } from '$lib/gen'
 	import { createEventDispatcher, getContext } from 'svelte'
 	import { Pen, RefreshCcw, Save } from 'lucide-svelte'
-	import DropdownV2 from '../../DropdownV2.svelte'
 	import type { FlowEditorContext } from '../types'
 	import { sendUserToast } from '$lib/utils'
 	import type { FlowBuilderWhitelabelCustomUi } from '$lib/components/custom_ui'
@@ -83,16 +82,13 @@
 			tag={module.value.tag}
 			on:change={(e) => dispatch('tagChange', e.detail)}
 		/>
-		<DropdownV2
-			size="sm"
-			placement="bottom-end"
-			items={[
-				{
-					displayName: 'Save to workspace',
-					icon: Save,
-					action: () => dispatch('createScriptFromInlineScript')
-				}
-			]}
+		<Button
+			unifiedSize="sm"
+			variant="subtle"
+			startIcon={{ icon: Save }}
+			on:click={() => dispatch('createScriptFromInlineScript')}
+			iconOnly={true}
+			title="Save to workspace"
 		/>
 	{/if}
 </div>