Minor visual bug fix