fix minor UI bugs