Fixed UI issue

Fixed UI issue