Update migration to use find_in_batches.