Fixed a display bug