diff --git a/multinode-demo/test/wallet-sanity.sh b/multinode-demo/test/wallet-sanity.sh
index 8a77cbe004..1e9d3374d6 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() {