Define the jsx namespace in the ./jsx-runtime module

README.md

@@ -267,7 +267,7 @@ Just like built-in modules, you can pass an arbitrary value to your custom modul
 Unlike built-in modules, we have no assumptions on what kind of values should be passed to custom modules. You have to augment `jsx.CustomModules` interface so that it will typecheck.
 
 ```ts
-declare module '@herp-inc/snabbdom-jsx' {
+declare module '@herp-inc/snabbdom-jsx/jsx-runtime' {
   namespace jsx {
     interface CustomModules {
       // Add your custom modules here

src/index.ts

@@ -1,18 +1,4 @@
-import type { Attrs, Classes, Dataset, On, Props } from 'snabbdom';
-
 import type Snabbdom from './jsx-runtime';
 
-declare namespace jsx {
-    interface CustomModules {
-        attrs: Attrs;
-        class: Classes;
-        dataset: Dataset;
-        props: Props;
-        on: On;
-        style: Snabbdom.Style;
-    }
-}
-
 // eslint-disable-next-line import/no-default-export
 export default Snabbdom;
-export type { jsx };

src/jsx-runtime.ts

@@ -1,8 +1,6 @@
 import type * as CSS from 'csstype';
 import type { Attrs, Classes, Dataset, Hooks, Key, On, Props, VNode, VNodeData } from 'snabbdom';
 
-import type { jsx } from '.';
-
 // eslint-disable-next-line @typescript-eslint/ban-types
 type Whatever = string & {};
 
@@ -1110,6 +1108,17 @@ declare namespace JSX {
     }
 }
 
+declare namespace jsx {
+    interface CustomModules {
+        attrs: Attrs;
+        class: Classes;
+        dataset: Dataset;
+        props: Props;
+        on: On;
+        style: Snabbdom.Style;
+    }
+}
+
 const isArrayChildren = (children: Snabbdom.Node): children is readonly Snabbdom.Node[] => Array.isArray(children);
 
 const kebab2camel = (kebab: string): string => {