LPS-51151 Fix unit test in a simple way, will refactor unit test later.