diff --git a/src/Umbraco.Web.UI.Client/src/packages/templating/stylesheets/tree/stylesheet-tree.repository.ts b/src/Umbraco.Web.UI.Client/src/packages/templating/stylesheets/tree/stylesheet-tree.repository.ts
index 03362dfbf7..dc3253ddc7 100644
--- a/src/Umbraco.Web.UI.Client/src/packages/templating/stylesheets/tree/stylesheet-tree.repository.ts
+++ b/src/Umbraco.Web.UI.Client/src/packages/templating/stylesheets/tree/stylesheet-tree.repository.ts
@@ -1,9 +1,13 @@
 import { UmbStylesheetTreeServerDataSource } from './stylesheet-tree.server.data-source.js';
 import { UMB_STYLESHEET_TREE_STORE_CONTEXT_TOKEN } from './stylesheet-tree.store.js';
+import { UmbStylesheetTreeItemModel, UmbStylesheetTreeRootModel } from './types.js';
 import { UmbTreeRepositoryBase } from '@umbraco-cms/backoffice/tree';
 import { UmbControllerHost } from '@umbraco-cms/backoffice/controller-api';
 
-export class UmbStylesheetTreeRepository extends UmbTreeRepositoryBase<any, any> {
+export class UmbStylesheetTreeRepository extends UmbTreeRepositoryBase<
+	UmbStylesheetTreeItemModel,
+	UmbStylesheetTreeRootModel
+> {
 	constructor(host: UmbControllerHost) {
 		super(host, UmbStylesheetTreeServerDataSource, UMB_STYLESHEET_TREE_STORE_CONTEXT_TOKEN);
 	}
@@ -11,7 +15,7 @@ export class UmbStylesheetTreeRepository extends UmbTreeRepositoryBase<any, any>
 	async requestTreeRoot() {
 		const data = {
 			path: null,
-			type: 'stylesheet-root',
+			entityType: 'stylesheet-root',
 			name: 'Stylesheets',
 			icon: 'icon-folder',
 			hasChildren: true,
diff --git a/src/Umbraco.Web.UI.Client/src/packages/templating/stylesheets/tree/types.ts b/src/Umbraco.Web.UI.Client/src/packages/templating/stylesheets/tree/types.ts
index b0fd9cd505..646cb5f854 100644
--- a/src/Umbraco.Web.UI.Client/src/packages/templating/stylesheets/tree/types.ts
+++ b/src/Umbraco.Web.UI.Client/src/packages/templating/stylesheets/tree/types.ts
@@ -1,5 +1,5 @@
-import type { UmbFileSystemTreeItemModel } from '@umbraco-cms/backoffice/tree';
+import type { UmbFileSystemTreeItemModel, UmbFileSystemTreeRootModel } from '@umbraco-cms/backoffice/tree';
 
 export interface UmbStylesheetTreeItemModel extends UmbFileSystemTreeItemModel {}
 // TODO: TREE STORE TYPE PROBLEM:
-export interface UmbStylesheetTreeRootModel extends UmbFileSystemTreeItemModel {}
+export interface UmbStylesheetTreeRootModel extends UmbFileSystemTreeRootModel {}