Small fixes to the UI.