fix the issue with __init__.py