1 files changed, 5 insertions, 3 deletions
diff --git a/lib/gitlab/background_migration/backfill_group_features.rb b/lib/gitlab/background_migration/backfill_group_features.rb
index 4c3af7be319..35b5282360f 100644
--- a/lib/gitlab/background_migration/backfill_group_features.rb
+++ b/lib/gitlab/background_migration/backfill_group_features.rb
@@ -4,19 +4,21 @@ module Gitlab
   module BackgroundMigration
     # Backfill group_features for an array of groups
     class BackfillGroupFeatures < ::Gitlab::BackgroundMigration::BatchedMigrationJob
-      def perform(batch_size)
+      job_arguments :batch_size
+
+      def perform
         each_sub_batch(
           operation_name: :upsert_group_features,
           batching_arguments: { order_hint: :type },
           batching_scope: ->(relation) { relation.where(type: 'Group') }
         ) do |sub_batch|
-          upsert_group_features(sub_batch, batch_size)
+          upsert_group_features(sub_batch)
         end
       end
 
       private
 
-      def upsert_group_features(relation, batch_size)
+      def upsert_group_features(relation)
         connection.execute(
           <<~SQL
           INSERT INTO group_features (group_id, created_at, updated_at)