KER-300: Fix unit test