Minor output fix