Actually fixing the migration script