Mini fix in migration