Fixed import in base __init__.py