Fixed a minor UI issue