diff --git a/lib/index.js b/lib/index.js
index 7696047..1e2a4a1 100644
--- a/lib/index.js
+++ b/lib/index.js
@@ -1,7 +1,7 @@
 /**
  * @import {Element, Nodes, Parents, Root} from 'hast'
  * @import {Root as MdastRoot} from 'mdast'
- * @import {ComponentProps, ElementType, ReactElement} from 'react'
+ * @import {ComponentType, JSX, ReactElement} from 'react'
  * @import {Options as RemarkRehypeOptions} from 'remark-rehype'
  * @import {BuildVisitor} from 'unist-util-visit'
  * @import {PluggableList, Processor} from 'unified'
@@ -29,7 +29,7 @@
 
 /**
  * @typedef {{
- *   [Key in Extract<ElementType, string>]?: ElementType<ComponentProps<Key> & ExtraProps>
+ *   [Key in keyof JSX.IntrinsicElements]?: ComponentType<JSX.IntrinsicElements[Key] & ExtraProps> | keyof JSX.IntrinsicElements
  * }} Components
  *   Map tag names to components.
  */