diff --git a/src/Umbraco.Web.UI.Client/src/packages/templating/templates/repository/template-item.store.ts b/src/Umbraco.Web.UI.Client/src/packages/templating/templates/repository/template-item.store.ts
index 1a167d099b..6f182fcea2 100644
--- a/src/Umbraco.Web.UI.Client/src/packages/templating/templates/repository/template-item.store.ts
+++ b/src/Umbraco.Web.UI.Client/src/packages/templating/templates/repository/template-item.store.ts
@@ -2,12 +2,11 @@ import { TemplateItemResponseModel } from '@umbraco-cms/backoffice/backend-api';
 import { UmbContextToken } from '@umbraco-cms/backoffice/context-api';
 import { UmbControllerHostElement } from '@umbraco-cms/backoffice/controller-api';
 import { UmbEntityItemStore } from '@umbraco-cms/backoffice/store';
-import { UmbArrayState } from '@umbraco-cms/backoffice/observable-api';
 
 /**
  * @export
  * @class UmbTemplateItemStore
- * @extends {UmbStoreBase}
+ * @extends {UmbEntityItemStore}
  * @description - Data Store for Template items
  */