Fix __init__.py