Fix migration script again