diff --git a/src/Umbraco.Web.UI.Client/src/mocks/data/media-type.data.ts b/src/Umbraco.Web.UI.Client/src/mocks/data/media-type.data.ts
index 4cf1363d3e..a45f091cba 100644
--- a/src/Umbraco.Web.UI.Client/src/mocks/data/media-type.data.ts
+++ b/src/Umbraco.Web.UI.Client/src/mocks/data/media-type.data.ts
@@ -1,7 +1,7 @@
 import type { MediaTypeDetails } from '../../packages/media/media-types/types.js';
 import { UmbEntityData } from './entity.data.js';
 import { createFolderTreeItem } from './utils.js';
-import { FolderTreeItemResponseModel, PagedFolderTreeItemResponseModel } from '@umbraco-cms/backoffice/backend-api';
+import { FolderTreeItemResponseModel, PagedMediaTreeItemResponseModel } from '@umbraco-cms/backoffice/backend-api';
 
 export const data: Array<MediaTypeDetails> = [
 	{
@@ -37,14 +37,14 @@ class UmbMediaTypeData extends UmbEntityData<MediaTypeDetails> {
 		super(data);
 	}
 
-	getTreeRoot(): PagedFolderTreeItemResponseModel {
+	getTreeRoot(): PagedMediaTreeItemResponseModel {
 		const items = this.data.filter((item) => item.parentId === null);
 		const treeItems = items.map((item) => createFolderTreeItem(item));
 		const total = items.length;
 		return { items: treeItems, total };
 	}
 
-	getTreeItemChildren(id: string): PagedFolderTreeItemResponseModel {
+	getTreeItemChildren(id: string): PagedMediaTreeItemResponseModel {
 		const items = this.data.filter((item) => item.parentId === id);
 		const treeItems = items.map((item) => createFolderTreeItem(item));
 		const total = items.length;
diff --git a/src/Umbraco.Web.UI.Client/src/mocks/data/media.data.ts b/src/Umbraco.Web.UI.Client/src/mocks/data/media.data.ts
index b28201d42c..fd24e3d456 100644
--- a/src/Umbraco.Web.UI.Client/src/mocks/data/media.data.ts
+++ b/src/Umbraco.Web.UI.Client/src/mocks/data/media.data.ts
@@ -2,7 +2,7 @@ import type { MediaDetails } from '../../packages/media/media/index.js';
 import { UmbEntityTreeData } from './entity-tree.data.js';
 import { UmbEntityData } from './entity.data.js';
 import { createContentTreeItem } from './utils.js';
-import { ContentTreeItemResponseModel, PagedContentTreeItemResponseModel } from '@umbraco-cms/backoffice/backend-api';
+import { ContentTreeItemResponseModel, PagedMediaTreeItemResponseModel } from '@umbraco-cms/backoffice/backend-api';
 
 export const data: Array<MediaDetails> = [
 	{
@@ -194,14 +194,14 @@ class UmbMediaData extends UmbEntityData<MediaDetails> {
 		super(data);
 	}
 
-	getTreeRoot(): PagedContentTreeItemResponseModel {
+	getTreeRoot(): PagedMediaTreeItemResponseModel {
 		const items = this.data.filter((item) => item.parentId === null);
 		const treeItems = items.map((item) => createContentTreeItem(item));
 		const total = items.length;
 		return { items: treeItems, total };
 	}
 
-	getTreeItemChildren(id: string): PagedContentTreeItemResponseModel {
+	getTreeItemChildren(id: string): PagedMediaTreeItemResponseModel {
 		const items = this.data.filter((item) => item.parentId === id);
 		const treeItems = items.map((item) => createContentTreeItem(item));
 		const total = items.length;