From b2c2c29bf7ca2bb52253042d90a5c71e88b1ac2d Mon Sep 17 00:00:00 2001
From: sakridge <sakridge@gmail.com>
Date: Tue, 3 May 2022 13:33:53 +0200
Subject: [PATCH] Fix quic mode with thinclient (#24890)

---
 bench-tps/src/main.rs | 6 +++---
 1 file changed, 3 insertions(+), 3 deletions(-)

diff --git a/bench-tps/src/main.rs b/bench-tps/src/main.rs
index 8c40930a2..627a89823 100644
--- a/bench-tps/src/main.rs
+++ b/bench-tps/src/main.rs
@@ -101,6 +101,9 @@ fn main() {
             do_bench_tps(client, cli_config, keypairs);
         }
         ExternalClientType::ThinClient => {
+            if *use_quic {
+                connection_cache::set_use_quic(true);
+            }
             let client = if let Ok(rpc_addr) = value_t!(matches, "rpc_addr", String) {
                 let rpc = rpc_addr.parse().unwrap_or_else(|e| {
                     eprintln!("RPC address should parse as socketaddr {:?}", e);
@@ -122,9 +125,6 @@ fn main() {
                             eprintln!("Failed to discover {} nodes: {:?}", num_nodes, err);
                             exit(1);
                         });
-                if *use_quic {
-                    connection_cache::set_use_quic(true);
-                }
                 if *multi_client {
                     let (client, num_clients) =
                         get_multi_client(&nodes, &SocketAddrSpace::Unspecified);