diff --git a/src/Umbraco.Web.UI.Client/src/packages/core/tree/types.ts b/src/Umbraco.Web.UI.Client/src/packages/core/tree/types.ts
index beb8e46f79..d0d18a96fc 100644
--- a/src/Umbraco.Web.UI.Client/src/packages/core/tree/types.ts
+++ b/src/Umbraco.Web.UI.Client/src/packages/core/tree/types.ts
@@ -1,14 +1,15 @@
 export interface UmbTreeItemModel {
-	type: string;
-	name: string;
-	hasChildren: boolean;
-	icon?: string;
+	type?: string; // TODO: remove option to be undefined when we have our own types
+	name?: string;
+	hasChildren?: boolean;
+	icon?: string | null;
+	parentId?: string | null;
 }
 
 export interface UmbEntityTreeItemModel extends UmbTreeItemModel {
-	id: string;
+	id?: string; // TODO: remove option to be undefined when server returns the same
 }
 
 export interface UmbFileSystemTreeItemModel {
-	path: string;
+	path?: string; // TODO: remove option to be undefined when server returns the same
 }