Fix UI tests errors