Minor UI fixes

Fixed inability to display