Fix the UI error