LPS-51165 Fix tests

cc/ @jorgeFerrer