Fix a bug in the output