this is able to load classes from `__init__.py`, which the previous code couldn't