diff --git a/specification/_global/search/_types/highlighting.ts b/specification/_global/search/_types/highlighting.ts
index 23d0c33b59..e0649e984c 100644
--- a/specification/_global/search/_types/highlighting.ts
+++ b/specification/_global/search/_types/highlighting.ts
@@ -20,7 +20,7 @@
 import { Field, Fields } from '@_types/common'
 import { integer } from '@_types/Numeric'
 import { QueryContainer } from '@_types/query_dsl/abstractions'
-import { Dictionary } from '@spec_utils/Dictionary'
+import { Dictionary, SingleKeyDictionary } from '@spec_utils/Dictionary'
 import { UserDefinedValue } from '@spec_utils/UserDefinedValue'
 
 export enum BoundaryScanner {
@@ -151,7 +151,9 @@ export class HighlightBase {
 
 export class Highlight extends HighlightBase {
   encoder?: HighlighterEncoder
-  fields: Dictionary<Field, HighlightField>
+  fields:
+    | SingleKeyDictionary<Field, HighlightField>
+    | SingleKeyDictionary<Field, HighlightField>[]
 }
 
 export enum HighlighterEncoder {
diff --git a/specification/_types/query_dsl/WeightedTokensQuery.ts b/specification/_types/query_dsl/WeightedTokensQuery.ts
index d09baa529c..4305ca7edb 100644
--- a/specification/_types/query_dsl/WeightedTokensQuery.ts
+++ b/specification/_types/query_dsl/WeightedTokensQuery.ts
@@ -18,7 +18,7 @@
  */
 
 import { float } from '@_types/Numeric'
-import { Dictionary } from '@spec_utils/Dictionary'
+import { SingleKeyDictionary } from '@spec_utils/Dictionary'
 import { QueryBase } from './abstractions'
 import { TokenPruningConfig } from './TokenPruningConfig'
 
@@ -27,7 +27,7 @@ import { TokenPruningConfig } from './TokenPruningConfig'
  */
 export class WeightedTokensQuery extends QueryBase {
   /** The tokens representing this query */
-  tokens: Dictionary<string, float>
+  tokens: SingleKeyDictionary<string, float>[]
   /** Token pruning configurations */
   pruning_config?: TokenPruningConfig
 }