small fix to output