fix an UI bug