fix UI error