diff --git a/yarn-project/p2p/src/testbench/worker_client_manager.ts b/yarn-project/p2p/src/testbench/worker_client_manager.ts index d6f3b250fa96..d0e3d3d1334f 100644 --- a/yarn-project/p2p/src/testbench/worker_client_manager.ts +++ b/yarn-project/p2p/src/testbench/worker_client_manager.ts @@ -483,7 +483,8 @@ class WorkerClientManager { }; this.processes[0].send(aggregatorCmd); - const result = await this.waitForBenchResult(0, config.timeoutMs + 30000); + const aggregatorBudgetMs = config.timeoutMs + BENCHMARK_CONSTANTS.MAX_PEER_WAIT_MS + 30000; + const result = await this.waitForBenchResult(0, aggregatorBudgetMs); this.logger.info( `Benchmark complete: fetched=${result.fetchedCount}/${config.txCount}, duration=${result.durationMs.toFixed(0)}ms, success=${result.success}`,