diff --git a/src/Umbraco.Web.UI.Client/libs/repository/repository-tree-data-source.interface.ts b/src/Umbraco.Web.UI.Client/libs/repository/repository-tree-data-source.interface.ts
index 5c8290dd11..5a62dff639 100644
--- a/src/Umbraco.Web.UI.Client/libs/repository/repository-tree-data-source.interface.ts
+++ b/src/Umbraco.Web.UI.Client/libs/repository/repository-tree-data-source.interface.ts
@@ -1,8 +1,7 @@
-import { EntityTreeItemResponseModel, PagedEntityTreeItemResponseModel } from '@umbraco-cms/backoffice/backend-api';
 import type { DataSourceResponse } from '@umbraco-cms/backoffice/repository';
 
-export interface RepositoryTreeDataSource<T = PagedEntityTreeItemResponseModel> {
-	getRootItems(): Promise<DataSourceResponse<T>>;
-	getChildrenOf(parentKey: string): Promise<DataSourceResponse<T>>;
-	getItems(key: Array<string>): Promise<DataSourceResponse<EntityTreeItemResponseModel[]>>;
+export interface RepositoryTreeDataSource<PagedItemsType, ItemsType> {
+	getRootItems(): Promise<DataSourceResponse<PagedItemsType>>;
+	getChildrenOf(parentUnique: string): Promise<DataSourceResponse<PagedItemsType>>;
+	getItems(unique: Array<string>): Promise<DataSourceResponse<Array<ItemsType>>>;
 }