Fixed minor UI issue