Fixed a slight UI bug.