Fixed Assertion error in base __init__.py