update to migration code