Fixed issue with the __init__.py file