LPS-44575 Update tests

cc/ @jorgeFerrer @topolik