Fixed minor bug of display