I don't think this should be in `__init__.py` can you put it in it's own file?