Fix docgen issue