fix doctest in Py3