Fix error in migration