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 cc63fa64dd..18a15749e9 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
@@ -88,7 +88,7 @@ export class UmbEntityData<T extends UmbEntityBase> extends UmbData<T> {
 		});
 	}
 
-	updateData(updateItem: T) {
+	updateData(updateItem: Partial<T>) {
 		const itemIndex = this.data.findIndex((item) => item.id === updateItem.id);
 		const item = this.data[itemIndex];
 		if (!item) return;