1 files changed, 5 insertions, 2 deletions
diff --git a/db/post_migrate/20170628080858_migrate_stage_id_reference_in_background.rb b/db/post_migrate/20170628080858_migrate_stage_id_reference_in_background.rb
index 1d95fc62c87..30849ea1361 100644
--- a/db/post_migrate/20170628080858_migrate_stage_id_reference_in_background.rb
+++ b/db/post_migrate/20170628080858_migrate_stage_id_reference_in_background.rb
@@ -12,9 +12,12 @@ class MigrateStageIdReferenceInBackground < ActiveRecord::Migration
   end
 
   def up
-    Build.where(stage_id: nil).in_batches(of: BATCH_SIZE) do |relation, index|
-      schedule = index * 5.minutes
+    index = 1
+
+    Build.where(stage_id: nil).in_batches(of: BATCH_SIZE) do |relation|
       jobs = relation.pluck(:id).map { |id| [MIGRATION, [id]] }
+      schedule = index * 5.minutes
+      index += 1
 
       BackgroundMigrationWorker.perform_bulk_in(schedule, jobs)
     end