1 files changed, 4 insertions, 8 deletions
diff --git a/lib/gitlab/background_migration/backfill_vulnerability_reads_cluster_agent.rb b/lib/gitlab/background_migration/backfill_vulnerability_reads_cluster_agent.rb
index 728b60f7a0e..0c41d6af209 100644
--- a/lib/gitlab/background_migration/backfill_vulnerability_reads_cluster_agent.rb
+++ b/lib/gitlab/background_migration/backfill_vulnerability_reads_cluster_agent.rb
@@ -10,16 +10,12 @@ module Gitlab
         vulnerability_reads.project_id = cluster_agents.project_id
       SQL
 
-      RELATION = ->(relation) do
-        relation
-          .where(report_type: 7)
-      end
+      CLUSTER_IMAGE_SCANNING_REPORT_TYPE = 7
+
+      scope_to ->(relation) { relation.where(report_type: CLUSTER_IMAGE_SCANNING_REPORT_TYPE) }
 
       def perform
-        each_sub_batch(
-          operation_name: :update_all,
-          batching_scope: RELATION
-        ) do |sub_batch|
+        each_sub_batch(operation_name: :update_all) do |sub_batch|
           sub_batch
             .joins(CLUSTER_AGENTS_JOIN)
             .update_all('casted_cluster_agent_id = CAST(vulnerability_reads.cluster_agent_id AS bigint)')