1 files changed, 3 insertions, 3 deletions
diff --git a/lib/gitlab/background_migration/sync_merge_requests_state_id.rb b/lib/gitlab/background_migration/sync_merge_requests_state_id.rb
index 5d92553f577..4cf8e993c52 100644
--- a/lib/gitlab/background_migration/sync_merge_requests_state_id.rb
+++ b/lib/gitlab/background_migration/sync_merge_requests_state_id.rb
@@ -9,8 +9,8 @@ module Gitlab
       def perform(start_id, end_id)
         Rails.logger.info("Merge Requests - Populating state_id: #{start_id} - #{end_id}")
 
-        reschedule_if_needed([start_id, end_id]) do
-          ActiveRecord::Base.connection.execute <<~SQL
+        reschedule_if_needed(start_id, end_id) do
+          execute_statement <<~SQL
             UPDATE merge_requests
             SET state_id =
               CASE state
@@ -27,7 +27,7 @@ module Gitlab
 
       private
 
-      def need_reschedule?
+      def should_reschedule?
         wait_for_deadtuple_vacuum?('issues')
       end
     end