Fixed problem with updating