fix UI

fix UI