diff --git a/src/Umbraco.Web.UI.Client/src/packages/documents/documents/entity-actions/publish.action.ts b/src/Umbraco.Web.UI.Client/src/packages/documents/documents/entity-actions/publish.action.ts
index 4c5dd7a0ac..cf8dbb3251 100644
--- a/src/Umbraco.Web.UI.Client/src/packages/documents/documents/entity-actions/publish.action.ts
+++ b/src/Umbraco.Web.UI.Client/src/packages/documents/documents/entity-actions/publish.action.ts
@@ -10,6 +10,7 @@ export class UmbPublishDocumentEntityAction extends UmbEntityActionBase<UmbDocum
 		const languageRepository = new UmbLanguageCollectionRepository(this._host);
 		const { data: languageData } = await languageRepository.requestCollection({});
 
+		// TODO: Not sure we need to use the Detail Repository for this, we might do just fine with the tree item model it self.
 		const documentRepository = new UmbDocumentDetailRepository(this._host);
 		const { data: documentData } = await documentRepository.requestByUnique(this.unique);
 
diff --git a/src/Umbraco.Web.UI.Client/src/packages/documents/documents/entity-actions/unpublish.action.ts b/src/Umbraco.Web.UI.Client/src/packages/documents/documents/entity-actions/unpublish.action.ts
index dc333eb654..db450cc88f 100644
--- a/src/Umbraco.Web.UI.Client/src/packages/documents/documents/entity-actions/unpublish.action.ts
+++ b/src/Umbraco.Web.UI.Client/src/packages/documents/documents/entity-actions/unpublish.action.ts
@@ -10,6 +10,7 @@ export class UmbUnpublishDocumentEntityAction extends UmbEntityActionBase<UmbDoc
 		const languageRepository = new UmbLanguageCollectionRepository(this._host);
 		const { data: languageData } = await languageRepository.requestCollection({});
 
+		// TODO: Not sure we need to use the Detail Repository for this, we might do just fine with the tree item model it self.
 		const documentRepository = new UmbDocumentDetailRepository(this._host);
 		const { data: documentData } = await documentRepository.requestByUnique(this.unique);