Improve readability of build stage id migration query

1 files changed, 2 insertions, 2 deletions

diff --git a/spec/migrations/migrate_stage_id_reference_in_background_spec.rb b/spec/migrations/migrate_stage_id_reference_in_background_spec.rb
index 1bd2c14b61c..63787d71233 100644
--- a/spec/migrations/migrate_stage_id_reference_in_background_spec.rb
+++ b/spec/migrations/migrate_stage_id_reference_in_background_spec.rb
@@ -58,11 +58,11 @@ describe MigrateStageIdReferenceInBackground, :migration, :sidekiq do
 
   it 'schedules background migrations' do
     Sidekiq::Testing.inline! do
-      expect(jobs.where(stage_id: nil)).to be_present
+      expect(jobs.where(stage_id: nil).count).to eq 5
 
       migrate!
 
-      expect(jobs.where(stage_id: nil)).to be_one
+      expect(jobs.where(stage_id: nil).count).to eq 1
     end
   end
 end