diff --git a/src/Umbraco.Web.UI.Client/libs/extensions-registry/models/tree-item.model.ts b/src/Umbraco.Web.UI.Client/libs/extensions-registry/models/tree-item.model.ts
index 8a50ad84ba..7785b849c5 100644
--- a/src/Umbraco.Web.UI.Client/libs/extensions-registry/models/tree-item.model.ts
+++ b/src/Umbraco.Web.UI.Client/libs/extensions-registry/models/tree-item.model.ts
@@ -1,6 +1,7 @@
+import { UmbTreeItemElement } from '../interfaces';
 import type { ManifestElement } from '.';
 
-export interface ManifestTreeItem extends ManifestElement {
+export interface ManifestTreeItem extends ManifestElement<UmbTreeItemElement> {
 	type: 'treeItem';
 	conditions: ConditionsTreeItem;
 }
diff --git a/src/Umbraco.Web.UI.Client/src/backoffice/templating/components/file-system-tree-item/file-system-tree-item.element.ts b/src/Umbraco.Web.UI.Client/src/backoffice/templating/components/file-system-tree-item/file-system-tree-item.element.ts
index 99f586c419..3236eb9676 100644
--- a/src/Umbraco.Web.UI.Client/src/backoffice/templating/components/file-system-tree-item/file-system-tree-item.element.ts
+++ b/src/Umbraco.Web.UI.Client/src/backoffice/templating/components/file-system-tree-item/file-system-tree-item.element.ts
@@ -6,6 +6,7 @@ import { UmbLitElement } from '@umbraco-cms/internal/lit-element';
 import { ManifestKind } from '@umbraco-cms/backoffice/extensions-registry';
 import { umbExtensionsRegistry } from '@umbraco-cms/backoffice/extensions-api';
 import { FileSystemTreeItemPresentationModel } from '@umbraco-cms/backoffice/backend-api';
+import { UmbTreeItemElement } from 'src/backoffice/shared/components/tree/tree-item/tree-item.element';
 
 // TODO: Move to separate file:
 const manifest: ManifestKind = {
@@ -21,7 +22,7 @@ const manifest: ManifestKind = {
 umbExtensionsRegistry.register(manifest);
 
 @customElement('umb-file-system-tree-item')
-export class UmbFileSystemTreeItemElement extends UmbLitElement {
+export class UmbFileSystemTreeItemElement extends UmbLitElement implements UmbTreeItemElement {
 	
 
 	private _item?: FileSystemTreeItemPresentationModel;