1 files changed, 6 insertions, 1 deletions
diff --git a/spec/features/user_sees_revert_modal_spec.rb b/spec/features/user_sees_revert_modal_spec.rb
index fdeee6a2808..ebb84a0d87f 100644
--- a/spec/features/user_sees_revert_modal_spec.rb
+++ b/spec/features/user_sees_revert_modal_spec.rb
@@ -12,6 +12,8 @@ RSpec.describe 'Merge request > User sees revert modal', :js, :sidekiq_might_not
     it 'shows the revert modal' do
       click_button('Revert')
 
+      wait_for_requests
+
       page.within('[data-testid="modal-commit"]') do
         expect(page).to have_content 'Revert this merge request'
       end
@@ -19,7 +21,6 @@ RSpec.describe 'Merge request > User sees revert modal', :js, :sidekiq_might_not
   end
 
   before do
-    stub_feature_flags(unbatch_graphql_queries: false)
     sign_in(user)
     visit(project_merge_request_path(project, merge_request))
 
@@ -27,6 +28,10 @@ RSpec.describe 'Merge request > User sees revert modal', :js, :sidekiq_might_not
       click_button 'Merge'
     end
 
+    wait_for_all_requests
+
+    page.refresh
+
     wait_for_requests
   end