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) {