KER-299: Fix unit test