corrected a little display bug