Fixed docgen issue