Fix some UI problems.