fixed UI bug