Update migration to remove redundancy