diff --git a/src/Umbraco.Web.UI.Client/src/mocks/data/partial-views.data.ts b/src/Umbraco.Web.UI.Client/src/mocks/data/partial-views.data.ts
index 7e98fe19d7..74f754477d 100644
--- a/src/Umbraco.Web.UI.Client/src/mocks/data/partial-views.data.ts
+++ b/src/Umbraco.Web.UI.Client/src/mocks/data/partial-views.data.ts
@@ -64,11 +64,7 @@ export const data: Array<FileSystemTreeItemPresentationModel> = [
 	},
 ];
 
-// Temp mocked database
-// TODO: all properties are optional in the server schema. I don't think this is correct.
-// eslint-disable-next-line @typescript-eslint/ban-ts-comment
-// @ts-ignore
-class UmbPartialViewsData extends UmbEntityData<FileSystemTreeItemPresentationModel> {
+class UmbPartialViewsTreeData extends UmbEntityData<FileSystemTreeItemPresentationModel> {
 	constructor() {
 		super(data);
 	}
@@ -93,4 +89,4 @@ class UmbPartialViewsData extends UmbEntityData<FileSystemTreeItemPresentationMo
 	}
 }
 
-export const umbPartialViewsData = new UmbPartialViewsData();
+export const umbPartialViewsTreeData = new UmbPartialViewsTreeData();
diff --git a/src/Umbraco.Web.UI.Client/src/mocks/handlers/partial-views.handlers.ts b/src/Umbraco.Web.UI.Client/src/mocks/handlers/partial-views.handlers.ts
index a4ab32e7c6..2e716e1dae 100644
--- a/src/Umbraco.Web.UI.Client/src/mocks/handlers/partial-views.handlers.ts
+++ b/src/Umbraco.Web.UI.Client/src/mocks/handlers/partial-views.handlers.ts
@@ -1,10 +1,10 @@
 const { rest } = window.MockServiceWorker;
-import { umbPartialViewsData } from '../data/partial-views.data.js';
+import { umbPartialViewsTreeData } from '../data/partial-views.data.js';
 import { umbracoPath } from '@umbraco-cms/backoffice/utils';
 
 export const handlers = [
 	rest.get(umbracoPath('/tree/partial-view/root'), (req, res, ctx) => {
-		const response = umbPartialViewsData.getTreeRoot();
+		const response = umbPartialViewsTreeData.getTreeRoot();
 		return res(ctx.status(200), ctx.json(response));
 	}),
 
@@ -12,7 +12,7 @@ export const handlers = [
 		const path = req.url.searchParams.get('path');
 		if (!path) return;
 
-		const response = umbPartialViewsData.getTreeItemChildren(path);
+		const response = umbPartialViewsTreeData.getTreeItemChildren(path);
 		return res(ctx.status(200), ctx.json(response));
 	}),
 
@@ -20,7 +20,7 @@ export const handlers = [
 		const paths = req.url.searchParams.getAll('paths');
 		if (!paths) return;
 
-		const items = umbPartialViewsData.getTreeItem(paths);
+		const items = umbPartialViewsTreeData.getTreeItem(paths);
 		return res(ctx.status(200), ctx.json(items));
 	}),
 ];