LPS-39277 Fix tests