LPS-33867 Fix the unit test