fix UI design issue