Fixed UI bugs.