LPS-39368 Fix unit test