Fix assertion rewriting in packages (``__init__.py``).