fixed an UI bug