diff --git a/src/Umbraco.Web.UI.Client/src/packages/core/utils/state-manager/state.manager.ts b/src/Umbraco.Web.UI.Client/src/packages/core/utils/state-manager/state.manager.ts
index a72be2b1dc..5b829e9512 100644
--- a/src/Umbraco.Web.UI.Client/src/packages/core/utils/state-manager/state.manager.ts
+++ b/src/Umbraco.Web.UI.Client/src/packages/core/utils/state-manager/state.manager.ts
@@ -1,3 +1,4 @@
+import { UmbControllerBase } from '@umbraco-cms/backoffice/class-api';
 import type { UmbControllerHost } from '@umbraco-cms/backoffice/controller-api';
 import { UmbArrayState } from '@umbraco-cms/backoffice/observable-api';
 
@@ -6,12 +7,12 @@ export interface UmbState {
 	message: string;
 }
 
-export class UmbStateManager<StateType extends UmbState = UmbState> {
+export class UmbStateManager<StateType extends UmbState = UmbState> extends UmbControllerBase {
 	protected _states = new UmbArrayState<StateType>([], (x) => x.unique);
 	public states = this._states.asObservable();
 
 	constructor(host: UmbControllerHost) {
-		//super(host);
+		super(host);
 	}
 
 	/**