Fix doctest in __init__.py