Fix UI issue.