Fix display problem