Fix typo in __init__.py

Thanks for open-sourcing this!