update migration code