fix some UI

fix some UI