LPS-30341 Fix unit test