diff --git a/src/Umbraco.Web.UI.Client/src/packages/templating/partial-views/components/input-partial-view/input-partial-view.context.ts b/src/Umbraco.Web.UI.Client/src/packages/templating/partial-views/components/input-partial-view/input-partial-view.context.ts
index 11c584246c..1dcff11475 100644
--- a/src/Umbraco.Web.UI.Client/src/packages/templating/partial-views/components/input-partial-view/input-partial-view.context.ts
+++ b/src/Umbraco.Web.UI.Client/src/packages/templating/partial-views/components/input-partial-view/input-partial-view.context.ts
@@ -1,13 +1,15 @@
+import { UmbPickerInputContext } from '@umbraco-cms/backoffice/picker-input';
+import type { UmbControllerHost } from '@umbraco-cms/backoffice/controller-api';
 import { UMB_PARTIAL_VIEW_ITEM_REPOSITORY_ALIAS } from '../../repository/item/index.js';
 import { UMB_PARTIAL_VIEW_PICKER_MODAL } from '../../partial-view-picker/index.js';
 import type { UmbPartialViewItemModel } from '../../types.js';
-import { UmbPickerInputContext } from '@umbraco-cms/backoffice/picker-input';
-import type { UmbControllerHost } from '@umbraco-cms/backoffice/controller-api';
+import type { UmbPartialViewTreeItemModel } from '../../tree/types.js';
 
-export class UmbPartialViewPickerContext extends UmbPickerInputContext<UmbPartialViewItemModel> {
+export class UmbPartialViewPickerContext extends UmbPickerInputContext<
+	UmbPartialViewItemModel,
+	UmbPartialViewTreeItemModel
+> {
 	constructor(host: UmbControllerHost) {
-		// eslint-disable-next-line @typescript-eslint/ban-ts-comment
-		// @ts-ignore
 		super(host, UMB_PARTIAL_VIEW_ITEM_REPOSITORY_ALIAS, UMB_PARTIAL_VIEW_PICKER_MODAL);
 	}
 }