Fixed import statement in __init__.py