fixed display bug