Fixed bug with updating.