Small fix in the output