Fix for documentation generation