Tweak to migration script.