Small fix to the output