correct the import statement in __init__.py