diff --git a/src/Umbraco.Web.UI.Client/src/backoffice/shared/components/tree/tree-item/tree-item.element.ts b/src/Umbraco.Web.UI.Client/src/backoffice/shared/components/tree/tree-item/tree-item.element.ts
index 462921d54d..d91f31c020 100644
--- a/src/Umbraco.Web.UI.Client/src/backoffice/shared/components/tree/tree-item/tree-item.element.ts
+++ b/src/Umbraco.Web.UI.Client/src/backoffice/shared/components/tree/tree-item/tree-item.element.ts
@@ -2,7 +2,7 @@ import { css, html } from 'lit';
 import { UUITextStyles } from '@umbraco-ui/uui-css/lib';
 import { customElement, property } from 'lit/decorators.js';
 import { UmbLitElement } from '@umbraco-cms/internal/lit-element';
-import { EntityTreeItemResponseModel } from 'libs/backend-api/src';
+import { TreeItemPresentationModel } from 'libs/backend-api/src';
 import { ManifestTreeItem } from 'libs/extensions-registry/tree-item.models';
 
 @customElement('umb-tree-item')
@@ -10,7 +10,7 @@ export class UmbTreeItemElement extends UmbLitElement {
 	static styles = [UUITextStyles, css``];
 
 	@property({ type: Object, attribute: false })
-	item?: EntityTreeItemResponseModel;
+	item?: TreeItemPresentationModel;
 
 	render() {
 		return html`<umb-extension-slot