minor fix UI