Fixed a display problem