Fix some UI stuff.