Fix display bug of highlight