Fix display bug during test