diff --git a/src/Umbraco.Web.UI.Client/src/packages/templating/scripts/components/input-script/input-script.context.ts b/src/Umbraco.Web.UI.Client/src/packages/templating/scripts/components/input-script/input-script.context.ts
index 37ccef1e72..cf3961fc41 100644
--- a/src/Umbraco.Web.UI.Client/src/packages/templating/scripts/components/input-script/input-script.context.ts
+++ b/src/Umbraco.Web.UI.Client/src/packages/templating/scripts/components/input-script/input-script.context.ts
@@ -1,13 +1,11 @@
 import { UMB_SCRIPT_ITEM_REPOSITORY_ALIAS } from '../../repository/item/index.js';
 import { UMB_SCRIPT_PICKER_MODAL } from '../../modals/script-picker-modal.token.js';
-import type { UmbScriptItemModel } from '../../index.js';
+import type { UmbScriptItemModel, UmbScriptTreeItemModel } from '../../index.js';
 import { UmbPickerInputContext } from '@umbraco-cms/backoffice/picker-input';
 import type { UmbControllerHost } from '@umbraco-cms/backoffice/controller-api';
 
-export class UmbScriptPickerContext extends UmbPickerInputContext<UmbScriptItemModel> {
+export class UmbScriptPickerContext extends UmbPickerInputContext<UmbScriptItemModel, UmbScriptTreeItemModel> {
 	constructor(host: UmbControllerHost) {
-		// eslint-disable-next-line @typescript-eslint/ban-ts-comment
-		// @ts-ignore
 		super(host, UMB_SCRIPT_ITEM_REPOSITORY_ALIAS, UMB_SCRIPT_PICKER_MODAL);
 	}
 }