minor fix to output