LPS-44662 Fix tests