fix display bug