fix UI issue