diff --git a/db-testing.yml b/db-testing.yml
index b27c02ad6faacd1c45599bb3b038bc5cff19ffe9..468359eebc0d4135ac92ac8fdf1a4fd69cd6aad5 100644
--- a/db-testing.yml
+++ b/db-testing.yml
@@ -85,8 +85,8 @@ db:migrations:
     - mkdir -p tmp/migration-testing/up
     - mkdir -p tmp/migration-testing/background_migrations
     - bundle exec rake gitlab:db:migration_testing:up | tee tmp/migration-testing/up/full-migration-output.log || true
-    - bundle exec rake gitlab:db:migration_testing:sample_background_migrations[900] | tee tmp/migration-testing/background_migrations/full-migration-output.log || true
-    - bundle exec rake gitlab:db:migration_testing:sample_batched_background_migrations[main,900] | tee tmp/migration-testing/background_migrations/full-batched-migration-output.log || true
+    - bundle exec rake gitlab:db:migration_testing:sample_background_migrations | tee tmp/migration-testing/background_migrations/full-migration-output.log || true
+    - bundle exec rake gitlab:db:migration_testing:sample_batched_background_migrations[main] | tee tmp/migration-testing/background_migrations/full-batched-migration-output.log || true
     - mv tmp/migration-testing/ ${CI_PROJECT_DIR}
     - bundle exec ruby list_migrations.rb > ${CI_PROJECT_DIR}/migration-testing/migrations.json
     - bundle exec ruby clone_details.rb > ${CI_PROJECT_DIR}/migration-testing/clone-details.json