Updated the update callback