fixing the migration script