diff --git a/db-testing.yml b/db-testing.yml
index 8809c66c9833efabaf68ada7f86e8a9d27a6b05b..0df88a506039a1f26afddb7b1570a62bd9d16343 100644
--- a/db-testing.yml
+++ b/db-testing.yml
@@ -77,8 +77,8 @@ image:redis:
     - 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 | 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
+    - 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
     - 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 ${CLONE_DETAILS_DB_NAME} > ${CI_PROJECT_DIR}/migration-testing/clone-details.json