Fixes documentation generation