Fix a small display bug