diff --git a/frontend/src/lib/components/InputTransformForm.svelte b/frontend/src/lib/components/InputTransformForm.svelte
index 6c90268d95..cad6c7a509 100644
--- a/frontend/src/lib/components/InputTransformForm.svelte
+++ b/frontend/src/lib/components/InputTransformForm.svelte
@@ -390,7 +390,7 @@
 					showSchemaExplorer
 				/>
 			{:else if arg.expr != undefined}
-				<div class="border rounded mt-2 border-gray-300">
+				<div class="border mt-2">
 					<SimpleEditor
 						bind:this={monaco}
 						bind:code={arg.expr}
diff --git a/frontend/src/lib/components/flows/content/FlowModuleComponent.svelte b/frontend/src/lib/components/flows/content/FlowModuleComponent.svelte
index d526b4a4a4..33f63f278b 100644
--- a/frontend/src/lib/components/flows/content/FlowModuleComponent.svelte
+++ b/frontend/src/lib/components/flows/content/FlowModuleComponent.svelte
@@ -102,7 +102,14 @@
 					previousId: undefined,
 					hasResume: false
 				},
-				extraLib: ''
+				extraLib: `
+				declare const error: {
+					message: string
+					name: string
+					stack: string
+				}
+				
+				`
 		  }
 		: getStepPropPicker(
 				$flowStateStore,
diff --git a/frontend/src/lib/script_helpers.ts b/frontend/src/lib/script_helpers.ts
index b3dc19bdd4..bc08b3152b 100644
--- a/frontend/src/lib/script_helpers.ts
+++ b/frontend/src/lib/script_helpers.ts
@@ -138,6 +138,15 @@ export async function main(message: string, name: string) {
 }
 `
 
+export const BUN_FAILURE_MODULE_CODE = `
+export async function main(message: string, name: string) {
+  const flow_id = process.env.WM_FLOW_JOB_ID
+  console.log("message", message)
+  console.log("name",name)
+  return { message, flow_id }
+}
+`
+
 export const POSTGRES_INIT_CODE = `-- to pin the database use '-- database f/your/path'
 -- $1 name1 = default arg
 -- $2 name2
@@ -403,10 +412,13 @@ export function initialCode(
 	} else if (language == 'bun') {
 		if (kind === 'approval') {
 			return BUN_INIT_CODE_APPROVAL
+		} else if (kind === 'failure') {
+			return BUN_FAILURE_MODULE_CODE
 		}
 		if (subkind === 'flow') {
 			return BUN_INIT_CODE_CLEAR
 		}
+
 		return BUN_INIT_CODE
 	} else {
 		if (kind === 'failure') {
diff --git a/frontend/src/routes/approve/[workspace]/[job]/[resume]/[hmac]/+page.svelte b/frontend/src/routes/approve/[workspace]/[job]/[resume]/[hmac]/+page.svelte
index 57feb9a954..0949ec8edf 100644
--- a/frontend/src/routes/approve/[workspace]/[job]/[resume]/[hmac]/+page.svelte
+++ b/frontend/src/routes/approve/[workspace]/[job]/[resume]/[hmac]/+page.svelte
@@ -208,6 +208,7 @@
 				<Alert type="warning" title="Adding a form to the approval page is an EE feature" />
 			{:else}
 				<SchemaForm
+					noVariablePicker
 					bind:isValid={valid}
 					schema={mergeSchema(schema, enum_payload)}
 					bind:args={default_payload}