fixed some UI