Fix a doctest for Python3