Data migration needs to be within dry_run block