diff --git a/src/Umbraco.Web.UI.Client/src/packages/core/tree/tree-data-source.interface.ts b/src/Umbraco.Web.UI.Client/src/packages/core/tree/tree-data-source.interface.ts
index 07a08c2d7f..0b4d673542 100644
--- a/src/Umbraco.Web.UI.Client/src/packages/core/tree/tree-data-source.interface.ts
+++ b/src/Umbraco.Web.UI.Client/src/packages/core/tree/tree-data-source.interface.ts
@@ -1,11 +1,18 @@
+import { UmbTreeItemModelBase } from './types.js';
 import type { UmbPagedData, DataSourceResponse } from '@umbraco-cms/backoffice/repository';
 import { UmbControllerHost } from '@umbraco-cms/backoffice/controller-api';
 
-export interface UmbTreeDataSourceConstructor<TreeItemType = any, TreeRootType = any> {
+export interface UmbTreeDataSourceConstructor<
+	TreeItemType extends UmbTreeItemModelBase,
+	TreeRootType extends UmbTreeItemModelBase,
+> {
 	new (host: UmbControllerHost): UmbTreeDataSource<TreeItemType, TreeRootType>;
 }
 
-export interface UmbTreeDataSource<TreeItemType = any, TreeRootType = any> {
+export interface UmbTreeDataSource<
+	TreeItemType extends UmbTreeItemModelBase,
+	TreeRootType extends UmbTreeItemModelBase,
+> {
 	getTreeRoot?(): Promise<DataSourceResponse<TreeRootType>>;
 	getRootItems(): Promise<DataSourceResponse<UmbPagedData<TreeItemType>>>;
 	getChildrenOf(parentUnique: string | null): Promise<DataSourceResponse<UmbPagedData<TreeItemType>>>;