Fix the documentation generation