Revert "Update __init__.py"

This reverts commit dc2b7735d5cb88369a241832ccfa96e92d5bd644.