[DF-4721] fix unit tests