[CAST-1751] Unit test fix