Fix display bug