fix display bug again