diff --git a/src/Umbraco.Web.UI.Client/src/mocks/data/document.data.ts b/src/Umbraco.Web.UI.Client/src/mocks/data/document.data.ts
index b2ab242576..5142811f32 100644
--- a/src/Umbraco.Web.UI.Client/src/mocks/data/document.data.ts
+++ b/src/Umbraco.Web.UI.Client/src/mocks/data/document.data.ts
@@ -11,6 +11,7 @@ import {
 	PagedDocumentTypeResponseModel,
 	PagedRecycleBinItemResponseModel,
 	PublishDocumentRequestModel,
+	PublishedStateModel,
 } from '@umbraco-cms/backoffice/backend-api';
 import { UMB_DOCUMENT_ENTITY_TYPE } from '@umbraco-cms/backoffice/document';
 
@@ -22,6 +23,7 @@ export const data: Array<DocumentResponseModel> = [
 				url: '/',
 			},
 		],
+		isTrashed: false,
 		templateId: null,
 		id: 'all-property-editors-document-id',
 		contentTypeId: 'all-property-editors-document-type-id',
@@ -359,6 +361,7 @@ export const data: Array<DocumentResponseModel> = [
 		templateId: null,
 		id: 'c05da24d-7740-447b-9cdc-bd8ce2172e38',
 		contentTypeId: '29643452-cff9-47f2-98cd-7de4b6807681',
+		isTrashed: false,
 		values: [
 			{
 				culture: null,
@@ -474,6 +477,7 @@ export const data: Array<DocumentResponseModel> = [
 		templateId: null,
 		id: 'fd56a0b5-01a0-4da2-b428-52773bfa9cc4',
 		contentTypeId: '29643452-cff9-47f2-98cd-7de4b6807681',
+		isTrashed: false,
 		values: [
 			{
 				culture: null,
@@ -546,6 +550,7 @@ export const data: Array<DocumentResponseModel> = [
 		templateId: null,
 		id: 'simple-document-id',
 		contentTypeId: 'simple-document-type-id',
+		isTrashed: false,
 		variants: [
 			{
 				state: ContentStateModel.DRAFT,
@@ -576,12 +581,25 @@ export const treeData: Array<DocumentTreeItemResponseModel> = [
 		noAccess: false,
 		isTrashed: false,
 		id: 'all-property-editors-document-id',
+		contentTypeId: 'all-property-editors-document-type-id',
 		isContainer: false,
 		parentId: null,
 		name: 'All property editors',
 		type: 'document',
 		icon: 'document',
 		hasChildren: false,
+		variants: [
+			{
+				name: 'All property editors',
+				culture: 'en-us',
+				state: PublishedStateModel.PUBLISHED,
+			},
+			{
+				name: 'All property editors',
+				culture: 'da-dk',
+				state: PublishedStateModel.PUBLISHED,
+			},
+		],
 	},
 	{
 		isProtected: false,
@@ -590,12 +608,40 @@ export const treeData: Array<DocumentTreeItemResponseModel> = [
 		noAccess: false,
 		isTrashed: false,
 		id: 'c05da24d-7740-447b-9cdc-bd8ce2172e38',
+		contentTypeId: '29643452-cff9-47f2-98cd-7de4b6807681',
 		isContainer: false,
 		parentId: null,
 		name: 'Article in english',
 		type: 'document',
 		icon: 'document',
 		hasChildren: true,
+		variants: [
+			{
+				state: PublishedStateModel.PUBLISHED,
+				culture: 'en-us',
+				name: 'Article in english',
+			},
+			{
+				state: PublishedStateModel.PUBLISHED,
+				culture: 'da-dk',
+				name: 'Artikel på Dansk',
+			},
+			{
+				state: PublishedStateModel.PUBLISHED,
+				culture: 'no-no',
+				name: 'Artikel på Norsk',
+			},
+			{
+				state: PublishedStateModel.PUBLISHED,
+				culture: 'es-es',
+				name: 'Articulo en ingles',
+			},
+			{
+				state: PublishedStateModel.PUBLISHED,
+				culture: 'pl-pl',
+				name: 'Artykuł w języku polskim',
+			},
+		],
 	},
 	{
 		isProtected: false,
@@ -604,26 +650,20 @@ export const treeData: Array<DocumentTreeItemResponseModel> = [
 		noAccess: false,
 		isTrashed: false,
 		id: 'fd56a0b5-01a0-4da2-b428-52773bfa9cc4',
+		contentTypeId: '29643452-cff9-47f2-98cd-7de4b6807681',
 		isContainer: false,
 		parentId: 'c05da24d-7740-447b-9cdc-bd8ce2172e38',
 		name: 'Blog post B',
 		type: 'document',
 		icon: 'document',
 		hasChildren: false,
-	},
-	{
-		name: 'Document 5',
-		type: 'document',
-		icon: 'document',
-		hasChildren: false,
-		id: 'f6n7a5b2-e7c1-463a-956bc-6ck5b9bdf447',
-		isContainer: false,
-		parentId: 'cdd30288-2d1c-41b4-89a9-61647b4a10d5',
-		noAccess: false,
-		isProtected: false,
-		isPublished: false,
-		isEdited: false,
-		isTrashed: false,
+		variants: [
+			{
+				state: PublishedStateModel.UNPUBLISHED,
+				culture: 'en-us',
+				name: 'Blog post B',
+			},
+		],
 	},
 	{
 		name: 'Simple',
@@ -631,6 +671,7 @@ export const treeData: Array<DocumentTreeItemResponseModel> = [
 		icon: 'document',
 		hasChildren: false,
 		id: 'simple-document-id',
+		contentTypeId: 'simple-document-type-id',
 		isContainer: false,
 		parentId: null,
 		noAccess: false,
@@ -638,20 +679,13 @@ export const treeData: Array<DocumentTreeItemResponseModel> = [
 		isPublished: false,
 		isEdited: false,
 		isTrashed: false,
-	},
-	{
-		name: 'Trashed',
-		type: 'document',
-		icon: 'document',
-		hasChildren: false,
-		id: 'trashed-document-id',
-		isContainer: false,
-		parentId: null,
-		noAccess: false,
-		isProtected: false,
-		isPublished: false,
-		isEdited: false,
-		isTrashed: true,
+		variants: [
+			{
+				state: PublishedStateModel.UNPUBLISHED,
+				culture: 'en-us',
+				name: 'Simple Document',
+			},
+		],
 	},
 ];