diff --git a/src/Umbraco.Web.UI.Client/src/mocks/data/entity-tree.data.ts b/src/Umbraco.Web.UI.Client/src/mocks/data/entity-tree.data.ts
index ce2c3710f7..c5436a0e15 100644
--- a/src/Umbraco.Web.UI.Client/src/mocks/data/entity-tree.data.ts
+++ b/src/Umbraco.Web.UI.Client/src/mocks/data/entity-tree.data.ts
@@ -9,15 +9,15 @@ export class UmbEntityTreeData<TreeItemType extends TreeItemPresentationModel> {
 		this.#detailDatabase = detailDatabase;
 	}
 
-	move(ids: Array<string>, destinationKey: string) {
-		const destinationItem = this.#detailDatabase.getById(destinationKey);
-		if (!destinationItem) throw new Error(`Destination item with key ${destinationKey} not found`);
+	move(ids: Array<string>, destinationId: string) {
+		const destinationItem = this.#detailDatabase.getById(destinationId);
+		if (!destinationItem) throw new Error(`Destination item with key ${destinationId} not found`);
 
 		const items = this.#detailDatabase.getByIds(ids);
 		const movedItems = items.map((item) => {
 			return {
 				...item,
-				parentId: destinationKey,
+				parentId: destinationId,
 			};
 		});
 
diff --git a/src/Umbraco.Web.UI.Client/src/mocks/data/entity.data.ts b/src/Umbraco.Web.UI.Client/src/mocks/data/entity.data.ts
index 4cfcce84a5..67e080f135 100644
--- a/src/Umbraco.Web.UI.Client/src/mocks/data/entity.data.ts
+++ b/src/Umbraco.Web.UI.Client/src/mocks/data/entity.data.ts
@@ -94,8 +94,8 @@ export class UmbEntityData<T extends UmbEntityBase> extends UmbData<T> {
 		return deletedKeys;
 	}
 
-	updateData(id: string, updateItem: T) {
-		const itemIndex = this.data.findIndex((item) => item.id === id);
+	updateData(updateItem: T) {
+		const itemIndex = this.data.findIndex((item) => item.id === updateItem.id);
 		const item = this.data[itemIndex];
 		if (!item) return;