Update to latest KB API