From 329aa0912243421ebe7c22edc098341cd67f5acd Mon Sep 17 00:00:00 2001
From: Mads Rasmussen <madsr@hey.com>
Date: Fri, 5 Apr 2024 21:23:35 +0200
Subject: [PATCH] use context token

---
 .../src/packages/core/entity/entity.context.ts                 | 3 ++-
 1 file changed, 2 insertions(+), 1 deletion(-)

diff --git a/src/Umbraco.Web.UI.Client/src/packages/core/entity/entity.context.ts b/src/Umbraco.Web.UI.Client/src/packages/core/entity/entity.context.ts
index f0b3620555..acb9abfc8b 100644
--- a/src/Umbraco.Web.UI.Client/src/packages/core/entity/entity.context.ts
+++ b/src/Umbraco.Web.UI.Client/src/packages/core/entity/entity.context.ts
@@ -1,3 +1,4 @@
+import { UMB_ENTITY_CONTEXT } from './entity.context-token.js';
 import { UmbContextBase } from '@umbraco-cms/backoffice/class-api';
 import type { UmbControllerHost } from '@umbraco-cms/backoffice/controller-api';
 import { UmbStringState } from '@umbraco-cms/backoffice/observable-api';
@@ -21,7 +22,7 @@ export class UmbEntityContext extends UmbContextBase<UmbEntityContext> {
 	 * @memberof UmbEntityContext
 	 */
 	constructor(host: UmbControllerHost) {
-		super(host, 'entity');
+		super(host, UMB_ENTITY_CONTEXT);
 	}
 
 	setEntityType(entityType: string | undefined) {