[WK2] Fix build break because of latest API changes