Update __init__.py

Reverting, need to add my change to a branch.