KER-278: Fixed the unit test