Minor fix for docgen