reverted the change to __init__.py