fix UI tests