diff --git a/src/Umbraco.Web.UI.Client/src/libs/extension-api/functions/create-extension-element.test.ts b/src/Umbraco.Web.UI.Client/src/libs/extension-api/functions/create-extension-element.test.ts
index fca6483987..5f2f8567df 100644
--- a/src/Umbraco.Web.UI.Client/src/libs/extension-api/functions/create-extension-element.test.ts
+++ b/src/Umbraco.Web.UI.Client/src/libs/extension-api/functions/create-extension-element.test.ts
@@ -2,16 +2,17 @@ import { expect } from '@open-wc/testing';
 import type { ManifestElement, ManifestElementAndApi } from '../types/index.js';
 import { createExtensionElement } from './create-extension-element.function.js';
 import { customElement } from '@umbraco-cms/backoffice/external/lit';
+import { UmbLitElement } from '@umbraco-cms/backoffice/lit-element';
 
 @customElement('umb-extension-api-true-test-element')
-class UmbExtensionApiTrueTestElement extends HTMLElement {
+class UmbExtensionApiTrueTestElement extends UmbLitElement {
 	isValidClassInstance() {
 		return true;
 	}
 }
 
 @customElement('umb-extension-api-false-test-element')
-class UmbExtensionApiFalseTestElement extends HTMLElement {
+class UmbExtensionApiFalseTestElement extends UmbLitElement {
 	isValidClassInstance() {
 		return false;
 	}