Fixed issue with updating