diff --git a/scribe/lib/scribe/ai/presentation/model/ai_scribe_menu_action.dart b/scribe/lib/scribe/ai/presentation/model/ai_scribe_menu_action.dart
index 26481c10c..fe67a4331 100644
--- a/scribe/lib/scribe/ai/presentation/model/ai_scribe_menu_action.dart
+++ b/scribe/lib/scribe/ai/presentation/model/ai_scribe_menu_action.dart
@@ -76,6 +76,8 @@ enum AIScribeMenuAction {
 
   String? getIcon(ImagePaths imagePaths) {
     switch (this) {
+      case AIScribeMenuAction.correctGrammar:
+        return imagePaths.icAiGrammar;
       case AIScribeMenuAction.improveMakeShorter:
         return imagePaths.icAiShorter;
       case AIScribeMenuAction.improveExpandContext: