Small fix for documentation generation