Fix a output bug