Fix the modify method