Fix the migration script.