test(+K): fix unit tests from refactor