__init__.py fix