1 files changed, 8 insertions, 1 deletions
diff --git a/spec/migrations/20230523101514_finalize_user_type_migration_spec.rb b/spec/migrations/20230523101514_finalize_user_type_migration_spec.rb
index abf3a506748..01c05c38098 100644
--- a/spec/migrations/20230523101514_finalize_user_type_migration_spec.rb
+++ b/spec/migrations/20230523101514_finalize_user_type_migration_spec.rb
@@ -5,7 +5,14 @@ require_migration!
 
 RSpec.describe FinalizeUserTypeMigration, feature_category: :devops_reports do
   it 'finalizes MigrateHumanUserType migration' do
-    expect(described_class).to be_finalize_background_migration_of('MigrateHumanUserType')
+    expect_next_instance_of(described_class) do |instance|
+      expect(instance).to receive(:ensure_batched_background_migration_is_finished).with(
+        job_class_name: 'MigrateHumanUserType',
+        table_name: :users,
+        column_name: :id,
+        job_arguments: []
+      )
+    end
 
     migrate!
   end