Minor output bug fix