fixed UI issue