minor __init__.py fix