fix the UI layout2