Update __init__.py

minor fix