LPS-52357 Update tests

cc @pablocp