Fixed the change in updating