Update __init__.py

mistake fix