Update __init__.py

Fixes a bug with setuptools