fix for update method