1 files changed, 3 insertions, 3 deletions
diff --git a/spec/services/merge_requests/reload_diffs_service_spec.rb b/spec/services/merge_requests/reload_diffs_service_spec.rb
index 77056cbe541..a6654989374 100644
--- a/spec/services/merge_requests/reload_diffs_service_spec.rb
+++ b/spec/services/merge_requests/reload_diffs_service_spec.rb
@@ -45,11 +45,11 @@ RSpec.describe MergeRequests::ReloadDiffsService, :use_clean_rails_memory_store_
         current_user
         merge_request
 
-        control_count = ActiveRecord::QueryRecorder.new do
+        control = ActiveRecord::QueryRecorder.new do
           subject.execute
-        end.count
+        end
 
-        expect { subject.execute }.not_to exceed_query_limit(control_count)
+        expect { subject.execute }.not_to exceed_query_limit(control)
       end
     end
   end