Fix __init__.py doctest