diff --git a/spec/schemas/_common.yaml b/spec/schemas/_common.yaml
index 8b22e9880..6450e9c6d 100644
--- a/spec/schemas/_common.yaml
+++ b/spec/schemas/_common.yaml
@@ -768,8 +768,10 @@ components:
         - wait_for
     WaitForActiveShards:
       oneOf:
-        - type: number
-        - $ref: '#/components/schemas/WaitForActiveShardOptions'
+        - title: count
+          type: integer
+        - title: option
+          $ref: '#/components/schemas/WaitForActiveShardOptions'
     WaitForActiveShardOptions:
       type: string
       enum: