Fixed tiny issue in __init__.py