diff --git a/frontend/src/lib/components/FlowBuilder.svelte b/frontend/src/lib/components/FlowBuilder.svelte
index 9033fd0939..1cec76a2a9 100644
--- a/frontend/src/lib/components/FlowBuilder.svelte
+++ b/frontend/src/lib/components/FlowBuilder.svelte
@@ -215,6 +215,7 @@
 						size="xs"
 						on:click={async () => {
 							$flowStore = undo(history, $flowStore)
+							$selectedIdStore = 'Input'
 						}}
 					>
 						<Undo size={14} />