diff --git a/specification/_types/query_dsl/WeightedTokensQuery.ts b/specification/_types/query_dsl/WeightedTokensQuery.ts
index 4305ca7edb..17f965510f 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 { SingleKeyDictionary } from '@spec_utils/Dictionary'
+import { Dictionary } 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: SingleKeyDictionary<string, float>[]
+  tokens: Dictionary<string, float> | Dictionary<string, float>[]
   /** Token pruning configurations */
   pruning_config?: TokenPruningConfig
 }