diff --git a/src/Umbraco.Web.UI.Client/src/backoffice/documents/documents/tree/tree-item/document-tree-item.element.ts b/src/Umbraco.Web.UI.Client/src/backoffice/documents/documents/tree/tree-item/document-tree-item.element.ts
index ad11c5badb..6f8c3699ed 100644
--- a/src/Umbraco.Web.UI.Client/src/backoffice/documents/documents/tree/tree-item/document-tree-item.element.ts
+++ b/src/Umbraco.Web.UI.Client/src/backoffice/documents/documents/tree/tree-item/document-tree-item.element.ts
@@ -4,11 +4,10 @@ import { customElement, property } from 'lit/decorators.js';
 import { UmbDocumentTreeItemContext } from './document-tree-item.context';
 import { UmbLitElement } from '@umbraco-cms/internal/lit-element';
 import { DocumentTreeItemResponseModel } from '@umbraco-cms/backoffice/backend-api';
+import { UmbTreeItemElement } from '@umbraco-cms/backoffice/extensions-registry';
 
 @customElement('umb-document-tree-item')
-export class UmbDocumentTreeItemElement extends UmbLitElement {
-	
-
+export class UmbDocumentTreeItemElement extends UmbLitElement implements UmbTreeItemElement {
 	private _item?: DocumentTreeItemResponseModel;
 	@property({ type: Object, attribute: false })
 	public get item() {
@@ -45,7 +44,7 @@ export class UmbDocumentTreeItemElement extends UmbLitElement {
 	#renderLabel() {
 		return html` <span id="label" slot="label">${this.item?.name}</span> `;
 	}
-	
+
 	static styles = [
 		UUITextStyles,
 		css`
diff --git a/src/Umbraco.Web.UI.Client/src/backoffice/shared/components/tree/entity-tree-item/entity-tree-item.element.ts b/src/Umbraco.Web.UI.Client/src/backoffice/shared/components/tree/entity-tree-item/entity-tree-item.element.ts
index a68e13eece..3bf3d6c4cd 100644
--- a/src/Umbraco.Web.UI.Client/src/backoffice/shared/components/tree/entity-tree-item/entity-tree-item.element.ts
+++ b/src/Umbraco.Web.UI.Client/src/backoffice/shared/components/tree/entity-tree-item/entity-tree-item.element.ts
@@ -4,7 +4,7 @@ import { customElement, property } from 'lit/decorators.js';
 import { UmbEntityTreeItemContext } from './entity-tree-item.context';
 import { UmbLitElement } from '@umbraco-cms/internal/lit-element';
 import { EntityTreeItemResponseModel } from '@umbraco-cms/backoffice/backend-api';
-import { ManifestKind } from '@umbraco-cms/backoffice/extensions-registry';
+import { ManifestKind, UmbTreeItemElement } from '@umbraco-cms/backoffice/extensions-registry';
 import { umbExtensionsRegistry } from '@umbraco-cms/backoffice/extensions-api';
 
 // TODO: Move to separate file:
@@ -21,7 +21,7 @@ const manifest: ManifestKind = {
 umbExtensionsRegistry.register(manifest);
 
 @customElement('umb-entity-tree-item')
-export class UmbEntityTreeItemElement extends UmbLitElement {
+export class UmbEntityTreeItemElement extends UmbLitElement implements UmbTreeItemElement {
 	
 
 	private _item?: EntityTreeItemResponseModel;