Update __init__.py

fix crash