diff --git a/src/index.d.ts b/src/index.d.ts
index 5c0bab3..a401390 100644
--- a/src/index.d.ts
+++ b/src/index.d.ts
@@ -43,6 +43,7 @@ type Options =
  *   shadow: true,
  *   mode: 'closed'
  * });
+ * const klass = register(PreactComponent, 'my-component');
  * ```
  */
 export default function register(
@@ -50,4 +51,4 @@ export default function register(
 	tagName?: string,
 	propNames?: string[],
 	options?: Options
-): void;
+): HTMLElement;
diff --git a/src/index.js b/src/index.js
index 287e6b7..5a0aac3 100644
--- a/src/index.js
+++ b/src/index.js
@@ -75,10 +75,12 @@ export default function register(Component, tagName, propNames, options) {
 		});
 	});
 
-	return customElements.define(
+	customElements.define(
 		tagName || Component.tagName || Component.displayName || Component.name,
 		PreactElement
 	);
+
+	return PreactElement;
 }
 
 function ContextProvider(props) {