Fix the UI