Fix recent bug in migration script.