trying to fix the display bug