Small fix in output