diff --git a/frontend/src/lib/components/ArgInput.svelte b/frontend/src/lib/components/ArgInput.svelte
index 8cad901f44..3ea5436a8e 100644
--- a/frontend/src/lib/components/ArgInput.svelte
+++ b/frontend/src/lib/components/ArgInput.svelte
@@ -51,6 +51,7 @@
 		| undefined = undefined
 	export let displayHeader = true
 	export let properties: { [name: string]: SchemaProperty } | undefined = undefined
+	export let nestedRequired: string[] | undefined = undefined
 	export let autofocus = false
 	export let compact = false
 	export let password = false
@@ -390,7 +391,7 @@
 						<SchemaForm
 							{disablePortal}
 							{disabled}
-							schema={{ properties, $schema: '', required: [], type: 'object' }}
+							schema={{ properties, $schema: '', required: nestedRequired ?? [], type: 'object' }}
 							bind:args={value}
 						/>
 					</div>
diff --git a/frontend/src/lib/components/InputTransformForm.svelte b/frontend/src/lib/components/InputTransformForm.svelte
index 4fa9a9b64a..684bb6d4ed 100644
--- a/frontend/src/lib/components/InputTransformForm.svelte
+++ b/frontend/src/lib/components/InputTransformForm.svelte
@@ -300,6 +300,7 @@
 					contentEncoding={schema.properties[argName].contentEncoding}
 					bind:itemsType={schema.properties[argName].items}
 					properties={schema.properties[argName].properties}
+					nestedRequired={schema.properties[argName].required}
 					displayHeader={false}
 					extra={argExtra}
 					{variableEditor}
diff --git a/frontend/src/lib/components/LightweightArgInput.svelte b/frontend/src/lib/components/LightweightArgInput.svelte
index 0458c6cf40..a657fcbb85 100644
--- a/frontend/src/lib/components/LightweightArgInput.svelte
+++ b/frontend/src/lib/components/LightweightArgInput.svelte
@@ -40,6 +40,7 @@
 		| undefined = undefined
 	export let displayHeader = true
 	export let properties: { [name: string]: SchemaProperty } | undefined = undefined
+	export let nestedRequired: string[] | undefined = undefined
 	export let extra: Record<string, any> = {}
 	export let displayType: boolean = true
 
@@ -277,7 +278,7 @@
 				{#if properties && Object.keys(properties).length > 0}
 					<div class="p-4 pl-8 border rounded w-full">
 						<LightweightSchemaForm
-							schema={{ properties, $schema: '', required: [], type: 'object' }}
+							schema={{ properties, $schema: '', required: nestedRequired ?? [], type: 'object' }}
 							bind:args={value}
 						/>
 					</div>
diff --git a/frontend/src/lib/components/LightweightSchemaForm.svelte b/frontend/src/lib/components/LightweightSchemaForm.svelte
index 81b10cc1fe..fdaceff84e 100644
--- a/frontend/src/lib/components/LightweightSchemaForm.svelte
+++ b/frontend/src/lib/components/LightweightSchemaForm.svelte
@@ -76,6 +76,7 @@
 				format={schema.properties[argName].format}
 				contentEncoding={schema.properties[argName].contentEncoding}
 				properties={schema.properties[argName].properties}
+				nestedRequired={schema.properties[argName].required}
 				itemsType={schema.properties[argName].items}
 				extra={schema.properties[argName]}
 				on:inputClicked
diff --git a/frontend/src/lib/components/ModulePreviewForm.svelte b/frontend/src/lib/components/ModulePreviewForm.svelte
index 5b91a52afe..3d45eb819e 100644
--- a/frontend/src/lib/components/ModulePreviewForm.svelte
+++ b/frontend/src/lib/components/ModulePreviewForm.svelte
@@ -92,6 +92,7 @@
 							format={schema.properties[argName].format}
 							contentEncoding={schema.properties[argName].contentEncoding}
 							properties={schema.properties[argName].properties}
+							nestedRequired={schema.properties[argName].required}
 							itemsType={schema.properties[argName].items}
 							extra={schema.properties[argName]}
 						/>
diff --git a/frontend/src/lib/components/SchemaEditor.svelte b/frontend/src/lib/components/SchemaEditor.svelte
index c9b5ff9884..ea2598ae1d 100644
--- a/frontend/src/lib/components/SchemaEditor.svelte
+++ b/frontend/src/lib/components/SchemaEditor.svelte
@@ -270,12 +270,12 @@
 </script>
 
 <div class="flex flex-col">
-	<div class="flex justify-between gap-x-2">
+	<div class="flex justify-between items-center gap-x-2">
 		<div>
 			<Button
 				variant="contained"
 				color="dark"
-				size="sm"
+				size={lightMode ? 'xs2' : 'sm'}
 				startIcon={{ icon: Plus }}
 				on:click={() => {
 					schemaModal.openDrawer(Object.assign({}, DEFAULT_PROPERTY))
@@ -288,6 +288,7 @@
 
 		<div class="flex items-center">
 			<Toggle
+				size={lightMode ? 'xs' : 'sm'}
 				on:change={() => switchTab()}
 				options={{
 					right: 'As JSON'
@@ -342,6 +343,7 @@
 		{:else}
 			<div class="border rounded p-2">
 				<SimpleEditor
+					small
 					bind:this={jsonEditor}
 					fixedOverflowWidgets={false}
 					on:change={() => {
diff --git a/frontend/src/lib/components/SchemaForm.svelte b/frontend/src/lib/components/SchemaForm.svelte
index 34f074cd70..86afb44e75 100644
--- a/frontend/src/lib/components/SchemaForm.svelte
+++ b/frontend/src/lib/components/SchemaForm.svelte
@@ -103,6 +103,7 @@
 								bind:format={schema.properties[argName].format}
 								contentEncoding={schema.properties[argName].contentEncoding}
 								properties={schema.properties[argName].properties}
+								nestedRequired={schema.properties[argName].required}
 								bind:itemsType={schema.properties[argName].items}
 								disabled={disabledArgs.includes(argName) || disabled}
 								{editableSchema}
@@ -131,6 +132,7 @@
 								format={schema.properties[argName].format}
 								contentEncoding={schema.properties[argName].contentEncoding}
 								properties={schema.properties[argName].properties}
+								nestedRequired={schema.properties[argName].required}
 								itemsType={schema.properties[argName].items}
 								disabled={disabledArgs.includes(argName) || disabled}
 								{compact}