Fix how update callback is called