update callback API as upstream changes