Small migration script improvements.