fixed some UI problem