Fix output in Python 3

Just fix the output to be correct in Python 3.