fix UI thing