Fixed a few UI related bugs.