Correct some UI elements