Did you mean to put this back into the `__init__.py` file?  I thought this was removed for the time being?