Fixed some UI bugs.