fixing display bug