wmt(4): Adapt to recent KPI changes
(cherry picked from commit 527b6d60f86953a40136abcf2f38cc4e70118954)