fix the display bug