Fix bug with updating.