diff --git a/src/Umbraco.Web.UI.Client/libs/context-api/provide/context-provider.controller.test.ts b/src/Umbraco.Web.UI.Client/libs/context-api/provide/context-provider.controller.test.ts
index 70ba73eacb..621ae25a1a 100644
--- a/src/Umbraco.Web.UI.Client/libs/context-api/provide/context-provider.controller.test.ts
+++ b/src/Umbraco.Web.UI.Client/libs/context-api/provide/context-provider.controller.test.ts
@@ -1,27 +1,26 @@
-import { expect, fixture, html } from '@open-wc/testing';
+import { expect, fixture, defineCE } from '@open-wc/testing';
 import { UmbContextConsumer } from '../consume/context-consumer';
 import { UmbContextProviderController } from './context-provider.controller';
-import { UmbControllerHostTestElement, UmbLitElement } from '@umbraco-cms/element';
+import { UmbLitElement } from '@umbraco-cms/element';
 
 class MyClass {
 	prop = 'value from provider';
 }
 
+class ControllerHostElement extends UmbLitElement {}
+const controllerHostElement = defineCE(ControllerHostElement);
+
 describe('UmbContextProviderController', () => {
 	let instance: MyClass;
 	let provider: UmbContextProviderController;
 	let element: UmbLitElement;
 
 	beforeEach(async () => {
-		element = await fixture(html`<umb-controller-host-test></umb-controller-host-test>`);
+		element = await fixture(`<${controllerHostElement}></${controllerHostElement}>`);
 		instance = new MyClass();
 		provider = new UmbContextProviderController(element, 'my-test-context', instance);
 	});
 
-	it('is defined with its own instance', () => {
-		expect(element).to.be.instanceOf(UmbControllerHostTestElement);
-	});
-
 	describe('Public API', () => {
 		describe('properties', () => {
 			it('has a unique property', () => {