fix in agent/__init__.py