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 c131d41fff..5c8290dd11 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,8 @@
 import { EntityTreeItemResponseModel, PagedEntityTreeItemResponseModel } from '@umbraco-cms/backoffice/backend-api';
 import type { DataSourceResponse } from '@umbraco-cms/backoffice/repository';
 
-export interface RepositoryTreeDataSource {
-	getRootItems(): Promise<DataSourceResponse<PagedEntityTreeItemResponseModel>>;
-	getChildrenOf(parentKey: string): Promise<DataSourceResponse<PagedEntityTreeItemResponseModel>>;
+export interface RepositoryTreeDataSource<T = PagedEntityTreeItemResponseModel> {
+	getRootItems(): Promise<DataSourceResponse<T>>;
+	getChildrenOf(parentKey: string): Promise<DataSourceResponse<T>>;
 	getItems(key: Array<string>): Promise<DataSourceResponse<EntityTreeItemResponseModel[]>>;
 }