fix some UI bug