diff --git a/packages/language-service/lib/features/provideDocumentSemanticTokens.ts b/packages/language-service/lib/features/provideDocumentSemanticTokens.ts
index be41118a..a4276460 100644
--- a/packages/language-service/lib/features/provideDocumentSemanticTokens.ts
+++ b/packages/language-service/lib/features/provideDocumentSemanticTokens.ts
@@ -35,6 +35,11 @@ export function register(context: ServiceContext) {
 			() => range!,
 			function* (map) {
 
+				for (const mapped of map.getGeneratedRanges(range!, isSemanticTokensEnabled)) {
+					yield mapped;
+					return;
+				}
+
 				let result: {
 					start: number;
 					end: number;