Fix doctests for py3