fix UI refresh issue