From 8d0d429acd4b0ecee88dece1689a740f58c8c405 Mon Sep 17 00:00:00 2001
From: Anatoly Yakovenko <aeyakovenko@gmail.com>
Date: Sun, 26 Aug 2018 23:31:26 -0700
Subject: [PATCH] update

---
 src/bin/bench-tps.rs | 6 +++++-
 1 file changed, 5 insertions(+), 1 deletion(-)

diff --git a/src/bin/bench-tps.rs b/src/bin/bench-tps.rs
index 477eee5806..b452effbb2 100644
--- a/src/bin/bench-tps.rs
+++ b/src/bin/bench-tps.rs
@@ -144,7 +144,11 @@ fn send_barrier_transaction(barrier_client: &mut ThinClient, last_id: &mut Hash,
 
             // Sanity check that the client balance is still 1
             let balance = barrier_client
-                .poll_get_balance(&id.pubkey())
+                .poll_balance_with_timeout(
+                    &id.pubkey(),
+                    &Duration::from_millis(100),
+                    &Duration::from_secs(10),
+                )
                 .expect("Failed to get balance");
             if balance != 1 {
                 panic!("Expected an account balance of 1 (balance: {}", balance);