From 02af24d27a423961b2ce700357036c7639f9b9f8 Mon Sep 17 00:00:00 2001
From: Fabian Boemer <fabian_boemer@apple.com>
Date: Tue, 5 Nov 2024 10:17:45 -0800
Subject: [PATCH] PIRProcessDatabase: fix response ciphertext count (#127)

---
 Sources/PIRProcessDatabase/ProcessDatabase.swift | 3 ++-
 1 file changed, 2 insertions(+), 1 deletion(-)

diff --git a/Sources/PIRProcessDatabase/ProcessDatabase.swift b/Sources/PIRProcessDatabase/ProcessDatabase.swift
index c42b29ea..f401f1fa 100644
--- a/Sources/PIRProcessDatabase/ProcessDatabase.swift
+++ b/Sources/PIRProcessDatabase/ProcessDatabase.swift
@@ -481,7 +481,8 @@ extension ProcessKeywordDatabase.ShardValidationResult {
             label: "keys"
         )
         descriptionDict["response size"] = try sizeString(byteCount: response.size(),
-                                                          count: response.ciphertexts.count, label: "ciphertexts")
+                                                          count: response.ciphertexts.flatMap { $0 }.count,
+                                                          label: "ciphertexts")
         descriptionDict["noise budget"] = String(format: "%.01f", noiseBudget)
 
         let runtimeString = computeTimes.sorted().map { runtime in