LPS-30336 Fix unit test