You should be importing directly from the relevant module, not using the `__init__.py`.