Fix rubocop offense in build stage migration classes

1 files changed, 2 insertions, 4 deletions

diff --git a/spec/migrations/schedule_build_stage_migration_spec.rb b/spec/migrations/schedule_build_stage_migration_spec.rb
index 79f00fdbcf0..b475e1fe7de 100644
--- a/spec/migrations/schedule_build_stage_migration_spec.rb
+++ b/spec/migrations/schedule_build_stage_migration_spec.rb
@@ -8,6 +8,8 @@ describe ScheduleBuildStageMigration, :migration do
   let(:jobs) { table(:ci_builds) }
 
   before do
+    stub_const("#{described_class}::BATCH", 1)
+
     ##
     # Dependencies
     #
@@ -24,10 +26,6 @@ describe ScheduleBuildStageMigration, :migration do
     jobs.create!(id: 4109, commit_id: 1, project_id: 123, stage_id: 1)
   end
 
-  before do
-    stub_const("#{described_class}::BATCH", 1)
-  end
-
   it 'schedules delayed background migrations in batches in bulk' do
     Sidekiq::Testing.fake! do
       Timecop.freeze do