Fix test failing on Python 3

Update the documentation too