LPS-48003 Fix tests

Thanks to @ealonso and @adolfopa :)