Fix a display bug