Reverted strange change that I don't remember making to update method.