diff --git a/frontend/src/lib/components/SavedInputs.svelte b/frontend/src/lib/components/SavedInputs.svelte
index 5baef88b1a..fbdb00a89a 100644
--- a/frontend/src/lib/components/SavedInputs.svelte
+++ b/frontend/src/lib/components/SavedInputs.svelte
@@ -317,7 +317,19 @@
 			<div class="h-full overflow-hidden min-h-0 flex flex-col justify-between">
 				<div class="w-full flex flex-col min-h-0 gap-2 px-2 py-2 grow">
 					<div class="text-sm font-extrabold">Preview</div>
-
+					<div class="w-full flex flex-col">
+						<Button
+							color="blue"
+							btnClasses="w-full"
+							size="sm"
+							spacingSize="xl"
+							on:click={() => selectArgs(selectedInput?.args)}
+							disabled={Object.keys(selectedInput?.args || {}).length === 0}
+						>
+							<ArrowLeftIcon class="w-4 h-4 mr-2" />
+							Use Input
+						</Button>
+					</div>
 					<div class="w-full min-h-0 grow overflow-auto">
 						{#if Object.keys(selectedInput?.args || {}).length > 0}
 							<div class="border overflow-auto h-full p-2">
@@ -330,21 +342,7 @@
 						{/if}
 					</div>
 				</div>
-
-				<div class="w-full flex flex-col px-2 pb-2">
-					<Button
-						color="blue"
-						btnClasses="w-full"
-						size="sm"
-						spacingSize="xl"
-						on:click={() => selectArgs(selectedInput?.args)}
-						disabled={Object.keys(selectedInput?.args || {}).length === 0}
-					>
-						<ArrowLeftIcon class="w-4 h-4 mr-2" />
-						Use Input
-					</Button>
-				</div>
-			</div>
-		</Pane>
+			</div></Pane
+		>
 	</Splitpanes>
 </div>
diff --git a/frontend/src/routes/(root)/(logged)/flows/get/[...path]/+page.svelte b/frontend/src/routes/(root)/(logged)/flows/get/[...path]/+page.svelte
index f6d8efac99..1d21dcb598 100644
--- a/frontend/src/routes/(root)/(logged)/flows/get/[...path]/+page.svelte
+++ b/frontend/src/routes/(root)/(logged)/flows/get/[...path]/+page.svelte
@@ -325,7 +325,9 @@
 				{isValid}
 				args={args ?? {}}
 				on:selected_args={(e) => {
-					runForm?.setArgs(JSON.parse(JSON.stringify(e.detail)))
+					const nargs = JSON.parse(JSON.stringify(e.detail))
+					runForm?.setArgs(nargs)
+					args = nargs
 				}}
 			/>
 		</svelte:fragment>
diff --git a/frontend/src/routes/(root)/(logged)/scripts/get/[...hash]/+page.svelte b/frontend/src/routes/(root)/(logged)/scripts/get/[...hash]/+page.svelte
index be27fe5452..31cc9532d0 100644
--- a/frontend/src/routes/(root)/(logged)/scripts/get/[...hash]/+page.svelte
+++ b/frontend/src/routes/(root)/(logged)/scripts/get/[...hash]/+page.svelte
@@ -515,7 +515,9 @@
 					{isValid}
 					{args}
 					on:selected_args={(e) => {
-						runForm?.setArgs(JSON.parse(JSON.stringify(e.detail)))
+						const nargs = JSON.parse(JSON.stringify(e.detail))
+						runForm?.setArgs(nargs)
+						args = nargs
 					}}
 				/>
 			{/if}