Fixed minor output bug