Fix documentation generation