fix minor output bug