Changed migration script