Fixed two small UI bugs