fix the ui