fix the __init__.py file