diff --git a/src/Umbraco.Web.UI.Client/src/packages/core/tree/entity-tree-item/entity-tree-item.context.ts b/src/Umbraco.Web.UI.Client/src/packages/core/tree/entity-tree-item/entity-tree-item.context.ts
index 8dbe286e3c..073d4fe44d 100644
--- a/src/Umbraco.Web.UI.Client/src/packages/core/tree/entity-tree-item/entity-tree-item.context.ts
+++ b/src/Umbraco.Web.UI.Client/src/packages/core/tree/entity-tree-item/entity-tree-item.context.ts
@@ -1,10 +1,9 @@
 import { UmbTreeItemContextBase } from '../tree-item-base/tree-item-base.context.js';
+import { UmbEntityTreeItemModel } from '../types.js';
 import { UmbControllerHostElement } from '@umbraco-cms/backoffice/controller-api';
-import { EntityTreeItemResponseModel } from '@umbraco-cms/backoffice/backend-api';
 
-// TODO get unique method from an entity repository static method
-export class UmbEntityTreeItemContext extends UmbTreeItemContextBase<EntityTreeItemResponseModel> {
+export class UmbEntityTreeItemContext extends UmbTreeItemContextBase<UmbEntityTreeItemModel> {
 	constructor(host: UmbControllerHostElement) {
-		super(host, (x: EntityTreeItemResponseModel) => x.id);
+		super(host, (x: UmbEntityTreeItemModel) => x.id);
 	}
 }
diff --git a/src/Umbraco.Web.UI.Client/src/packages/core/tree/entity-tree-item/entity-tree-item.element.ts b/src/Umbraco.Web.UI.Client/src/packages/core/tree/entity-tree-item/entity-tree-item.element.ts
index 3524ab5082..97b9abbc63 100644
--- a/src/Umbraco.Web.UI.Client/src/packages/core/tree/entity-tree-item/entity-tree-item.element.ts
+++ b/src/Umbraco.Web.UI.Client/src/packages/core/tree/entity-tree-item/entity-tree-item.element.ts
@@ -1,8 +1,8 @@
+import { UmbEntityTreeItemModel } from '../types.js';
 import { UmbEntityTreeItemContext } from './entity-tree-item.context.js';
 import { css, html, nothing, customElement, property } from '@umbraco-cms/backoffice/external/lit';
 import { UmbTextStyles } from '@umbraco-cms/backoffice/style';
 import { UmbLitElement } from '@umbraco-cms/internal/lit-element';
-import { EntityTreeItemResponseModel } from '@umbraco-cms/backoffice/backend-api';
 import {
 	UmbBackofficeManifestKind,
 	UmbTreeItemElement,
@@ -24,12 +24,12 @@ umbExtensionsRegistry.register(manifest);
 
 @customElement('umb-entity-tree-item')
 export class UmbEntityTreeItemElement extends UmbLitElement implements UmbTreeItemElement {
-	private _item?: EntityTreeItemResponseModel;
+	private _item?: UmbEntityTreeItemModel;
 	@property({ type: Object, attribute: false })
 	public get item() {
 		return this._item;
 	}
-	public set item(value: EntityTreeItemResponseModel | undefined) {
+	public set item(value: UmbEntityTreeItemModel | undefined) {
 		this._item = value;
 		this.#context.setTreeItem(value);
 	}