diff --git a/src/Umbraco.Web.UI.Client/devops/plop/templates/property-editor-ui/element.ts.hbs b/src/Umbraco.Web.UI.Client/devops/plop/templates/property-editor-ui/element.ts.hbs
index d33b4a38fa..304216e13f 100644
--- a/src/Umbraco.Web.UI.Client/devops/plop/templates/property-editor-ui/element.ts.hbs
+++ b/src/Umbraco.Web.UI.Client/devops/plop/templates/property-editor-ui/element.ts.hbs
@@ -1,12 +1,13 @@
-import { html, LitElement } from 'lit';
-import { UUITextStyles } from '@umbraco-ui/uui-css/lib';
+import { html } from 'lit';
 import { customElement, property } from 'lit/decorators.js';
+import { UUITextStyles } from '@umbraco-ui/uui-css/lib';
+import { UmbLitElement } from '@umbraco-cms/element';
 
 /**
  * @element {{ extensionTagName extensionType name }}
  */
 @customElement('{{ extensionTagName extensionType name }}')
-export class {{className extensionType name }} extends LitElement {
+export class {{className extensionType name }} extends UmbLitElement {
 	static styles = [UUITextStyles];
 
 	@property()