Fix wrong migration code