LPS-45935 Fix tests

cc @mdelapenya @izaera