Fixes the documentation generation