From 16cf31c3a3634fe70764a62462218b8e0a382362 Mon Sep 17 00:00:00 2001
From: Rob Walker <rob@solana.com>
Date: Thu, 16 Aug 2018 15:11:58 -0700
Subject: [PATCH] fix #990

---
 multinode-demo/test/wallet-sanity.sh | 23 +++++------------------
 1 file changed, 5 insertions(+), 18 deletions(-)

diff --git a/multinode-demo/test/wallet-sanity.sh b/multinode-demo/test/wallet-sanity.sh
index 8a77cbe00..1e9d3374d 100755
--- a/multinode-demo/test/wallet-sanity.sh
+++ b/multinode-demo/test/wallet-sanity.sh
@@ -18,26 +18,13 @@ fi
 garbage_address=vS3ngn1TfQmpsW1Z4NkLuqNAQFF3dYQw8UZ6TCx9bmq
 
 check_balance_output() {
-  # TODO go back to a single expected_output once snaps and wallet
-  #   are in sync
-  #
-  #   declare expected_output="$1"
-  #   exec 42>&1
-  #   output=$($wallet balance | tee >(cat - >&42))
-  #   if [[ ! "$output" =~ $expected_output ]]; then
-  #     echo "Balance is incorrect.  Expected: $expected_output"
-  #     exit 1
-  #   fi
+  declare expected_output="$1"
   exec 42>&1
   output=$($wallet balance | tee >(cat - >&42))
-
-  for expected_output in "$@"; do
-    if [[ "$output" =~ $expected_output ]]; then
-        return 0
-    fi
-  done
-  echo "Balance is incorrect.  Expected: $expected_output"
-  exit 1
+  if [[ ! "$output" =~ $expected_output ]]; then
+    echo "Balance is incorrect.  Expected: $expected_output"
+    exit 1
+  fi
 }
 
 pay_and_confirm() {