fix two small UI bugs