Fixed a display issue