Fixed some UI