diff --git a/apps/web/src/features/ai/AISidebar.tsx b/apps/web/src/features/ai/AISidebar.tsx
index c27b1a2..50ade10 100644
--- a/apps/web/src/features/ai/AISidebar.tsx
+++ b/apps/web/src/features/ai/AISidebar.tsx
@@ -1470,6 +1470,7 @@ export function AISidebar({ fullPage = false }: AISidebarProps) {
                           title="Click to highlight this line on the writing screen"
                           onClick={(e) => {
                             e.stopPropagation();
+                            setFocusedIndex(idx);
                             handleHighlightLine(sg.lineNumber ?? 1);
                           }}
                         >