Update __init__.py

fix in init file