diff --git a/src/Umbraco.Web.UI.Client/src/core/extensions-api/registry/extension.registry.ts b/src/Umbraco.Web.UI.Client/src/core/extensions-api/registry/extension.registry.ts
index 8ebdbdafe5..2d8e124baf 100644
--- a/src/Umbraco.Web.UI.Client/src/core/extensions-api/registry/extension.registry.ts
+++ b/src/Umbraco.Web.UI.Client/src/core/extensions-api/registry/extension.registry.ts
@@ -19,6 +19,7 @@ import type {
 	ManifestHeaderApp,
 	ManifestCollectionView,
 	ManifestCollectionBulkAction,
+	AliasManifestTypeMap,
 } from '../../models';
 import { hasDefaultExport } from '../has-default-export.function';
 import { loadExtension } from '../load-extension.function';
@@ -75,8 +76,8 @@ export class UmbExtensionRegistry {
 		return this.extensions.pipe(map((dataTypes) => dataTypes.find((extension) => extension.alias === alias) || null));
 	}
 
-	getByTypeAndAlias(type: string, alias: string) {
-		return this.extensionsOfType(type).pipe(map((extensions) => extensions.find((extension) => extension.alias === alias) || null));
+	getByTypeAndAlias<Key extends keyof AliasManifestTypeMap>(type: Key, alias: string) {
+		return this.extensionsOfType<AliasManifestTypeMap[Key]>(type).pipe(map((extensions) => extensions.find((extension) => extension.alias === alias) || null));
 	}
 
 	// TODO: implement unregister of extension