diff --git a/src/lib/components/Editor.svelte b/src/lib/components/Editor.svelte
index da84c5c..20166e8 100644
--- a/src/lib/components/Editor.svelte
+++ b/src/lib/components/Editor.svelte
@@ -4906,6 +4906,10 @@
 				aiSelectionTo = end;
 				aiSelectedText = selected;
 				aiWholeNote = false;
+				requestAnimationFrame(() => {
+					sourceElement?.focus();
+					sourceElement?.setSelectionRange(start, end);
+				});
 			} else {
 				const fullMd = restoreTitleH1(sourceContent);
 				if (!fullMd.trim()) {