Rolled back changes to __init__.py