Fix minor UI issue