small tweaks to migration