Fixing the migration script.