fix UI bugs