Try to fix migration script