diff --git a/specification/enrich/_types/Policy.ts b/specification/enrich/_types/Policy.ts
index aad7b99c37..20b19df3de 100644
--- a/specification/enrich/_types/Policy.ts
+++ b/specification/enrich/_types/Policy.ts
@@ -18,6 +18,7 @@
  */
 
 import { Field, Fields, Indices, Name } from '@_types/common'
+import { QueryContainer } from '@_types/query_dsl/abstractions'
 import { SingleKeyDictionary } from '@spec_utils/Dictionary'
 
 export class Summary {
@@ -34,7 +35,7 @@ export class Policy {
   enrich_fields: Fields
   indices: Indices
   match_field: Field
-  query?: string
+  query?: QueryContainer | QueryContainer[]
   name?: Name
   elasticsearch_version?: string
 }