fix the UI layout