diff --git a/app/src/code_review/code_review_view.rs b/app/src/code_review/code_review_view.rs
index 93e4e193..11f834d4 100644
--- a/app/src/code_review/code_review_view.rs
+++ b/app/src/code_review/code_review_view.rs
@@ -7025,15 +7025,6 @@ impl CodeReviewView {
 
         let has_changes = matches!(self.state(), CodeReviewViewState::Loaded(loaded) if !loaded.to_diff_stats().has_no_changes());
 
-        if FeatureFlag::DiscardPerFileAndAllChanges.is_enabled() && has_changes {
-            items.push(
-                MenuItemFields::new("Discard all")
-                    .with_icon(Icon::ReverseLeft)
-                    .with_on_select_action(CodeReviewAction::ShowDiscardConfirmDialog(None))
-                    .into_item(),
-            );
-        }
-
         let is_ai_enabled = AISettings::as_ref(ctx).is_any_ai_enabled(ctx);
         if is_ai_enabled && FeatureFlag::DiffSetAsContext.is_enabled() && has_changes {
             items.push(
@@ -7060,6 +7051,15 @@ impl CodeReviewView {
             );
         }
 
+        if FeatureFlag::DiscardPerFileAndAllChanges.is_enabled() && has_changes {
+            items.push(
+                MenuItemFields::new("Discard all")
+                    .with_icon(Icon::ReverseLeft)
+                    .with_on_select_action(CodeReviewAction::ShowDiscardConfirmDialog(None))
+                    .into_item(),
+            );
+        }
+
         items
     }