Added fix for the update bug.