fix UI break issue