Fix a couple visual bugs

See #211