Fixed the UI issue