Restore the update method