Fixed failing test for kpi refactor.