fixed the display bug