LPS-35561 Fix unit test