diff --git a/src/Umbraco.Web.UI.Client/src/packages/core/tree/data/tree-data-source.interface.ts b/src/Umbraco.Web.UI.Client/src/packages/core/tree/data/tree-data-source.interface.ts
index 231fab39b7..366855833a 100644
--- a/src/Umbraco.Web.UI.Client/src/packages/core/tree/data/tree-data-source.interface.ts
+++ b/src/Umbraco.Web.UI.Client/src/packages/core/tree/data/tree-data-source.interface.ts
@@ -1,4 +1,4 @@
-import type { UmbTreeItemModelBase } from '../types.js';
+import type { UmbUniqueTreeItemModel } from '../types.js';
 import type {
 	UmbTreeAncestorsOfRequestArgs,
 	UmbTreeChildrenOfRequestArgs,
@@ -13,7 +13,7 @@ import type { UmbControllerHost } from '@umbraco-cms/backoffice/controller-api';
  * @interface UmbTreeDataSourceConstructor
  * @template TreeItemType
  */
-export interface UmbTreeDataSourceConstructor<TreeItemType extends UmbTreeItemModelBase> {
+export interface UmbTreeDataSourceConstructor<TreeItemType extends UmbUniqueTreeItemModel> {
 	new (host: UmbControllerHost): UmbTreeDataSource<TreeItemType>;
 }
 
@@ -23,7 +23,7 @@ export interface UmbTreeDataSourceConstructor<TreeItemType extends UmbTreeItemMo
  * @interface UmbTreeDataSource
  * @template TreeItemType
  */
-export interface UmbTreeDataSource<TreeItemType extends UmbTreeItemModelBase> {
+export interface UmbTreeDataSource<TreeItemType extends UmbUniqueTreeItemModel> {
 	/**
 	 * Gets the root items of the tree.
 	 * @return {*}  {Promise<UmbDataSourceResponse<UmbPagedModel<TreeItemType>>>}
diff --git a/src/Umbraco.Web.UI.Client/src/packages/core/tree/data/tree-server-data-source-base.ts b/src/Umbraco.Web.UI.Client/src/packages/core/tree/data/tree-server-data-source-base.ts
index b248273b5a..eeffac2cdb 100644
--- a/src/Umbraco.Web.UI.Client/src/packages/core/tree/data/tree-server-data-source-base.ts
+++ b/src/Umbraco.Web.UI.Client/src/packages/core/tree/data/tree-server-data-source-base.ts
@@ -1,4 +1,4 @@
-import type { UmbTreeItemModelBase } from '../types.js';
+import type { UmbTreeItemModelBase, UmbUniqueTreeItemModel } from '../types.js';
 import type { UmbTreeDataSource } from './tree-data-source.interface.js';
 import type {
 	UmbTreeAncestorsOfRequestArgs,
@@ -27,7 +27,7 @@ export interface UmbTreeServerDataSourceBaseArgs<
  */
 export abstract class UmbTreeServerDataSourceBase<
 	ServerTreeItemType extends { hasChildren: boolean },
-	ClientTreeItemType extends UmbTreeItemModelBase,
+	ClientTreeItemType extends UmbUniqueTreeItemModel,
 > implements UmbTreeDataSource<ClientTreeItemType>
 {
 	#host;