diff --git a/src/ui/ai_chat_dialog.rs b/src/ui/ai_chat_dialog.rs
index 68f60e3..dffdf3e 100644
--- a/src/ui/ai_chat_dialog.rs
+++ b/src/ui/ai_chat_dialog.rs
@@ -232,6 +232,8 @@ impl AiChatDialog {
             imp.suggestion_list_box.append(&button);
         }
 
+        imp.message_entry.grab_focus();
+
         this
     }