diff --git a/signal-server-openapi.yaml b/signal-server-openapi.yaml
index 9653f6b2b..68f91163b 100644
--- a/signal-server-openapi.yaml
+++ b/signal-server-openapi.yaml
@@ -1587,6 +1587,10 @@ paths:
           enum:
           - ACI
           - PNI
+      - name: User-Agent
+        in: header
+        schema:
+          type: string
       responses:
         "200":
           description: Body contains the number of available one-time prekeys for
@@ -1594,7 +1598,17 @@ paths:
           content:
             application/json:
               schema:
-                $ref: '#/components/schemas/PreKeyCount'
+                type: object
+                properties:
+                  completedExceptionally:
+                    type: boolean
+                  numberOfDependents:
+                    type: integer
+                    format: int32
+                  done:
+                    type: boolean
+                  cancelled:
+                    type: boolean
         "401":
           description: Account authentication check failed.
       security:
@@ -1619,6 +1633,10 @@ paths:
           - PNI
           - aci
           - pni
+      - name: User-Agent
+        in: header
+        schema:
+          type: string
       responses:
         "200":
           description: Indicates that new keys were successfully stored.
@@ -3951,19 +3969,6 @@ components:
           items:
             type: string
             format: byte
-    PreKeyCount:
-      type: object
-      properties:
-        count:
-          type: integer
-          description: the number of stored unsigned elliptic-curve prekeys for this
-            device
-          format: int32
-        pqCount:
-          type: integer
-          description: the number of stored one-time post-quantum prekeys for this
-            device
-          format: int32
     SetKeysRequest:
       type: object
       properties: