Minor fix of display