changed import in __init__.py