From e421aaad0754cd10e4aa9c94a90dda8b43f2eb6d Mon Sep 17 00:00:00 2001
From: Kevin Lewi <klewi@fb.com>
Date: Thu, 21 Jan 2021 22:09:34 -0800
Subject: [PATCH] Fixing expect script for digital locker example (#118)

---
 examples/digital_locker.rs | 4 ++--
 src/map_to_curve.rs        | 3 +--
 2 files changed, 3 insertions(+), 4 deletions(-)

diff --git a/examples/digital_locker.rs b/examples/digital_locker.rs
index 7897de4..9617f31 100644
--- a/examples/digital_locker.rs
+++ b/examples/digital_locker.rs
@@ -224,13 +224,13 @@ fn main() {
                         let locker_index: usize = match locker.parse() {
                             Ok(index) => index,
                             Err(_) => {
-                                println!("Could not find locker number");
+                                println!("Error: Could not find locker number");
                                 continue;
                             }
                         };
 
                         if locker_index >= registered_lockers.len() {
-                            println!("Could not find locker number");
+                            println!("Error: Could not find locker number");
                             continue;
                         }
 
diff --git a/src/map_to_curve.rs b/src/map_to_curve.rs
index b5b58b1..3b28751 100644
--- a/src/map_to_curve.rs
+++ b/src/map_to_curve.rs
@@ -88,8 +88,7 @@ pub(crate) fn expand_message_xmd<H: Hash>(
     let l_i_b_str = i2osp(len_in_bytes, 2);
     let msg_prime = [&z_pad, msg, &l_i_b_str, &i2osp(0, 1), &dst_prime].concat();
 
-    let mut b: Vec<Vec<u8>> = Vec::new();
-    b.push(H::digest(&msg_prime).to_vec()); // b[0]
+    let mut b: Vec<Vec<u8>> = vec![H::digest(&msg_prime).to_vec()]; // b[0]
 
     let mut h = H::new();
     h.update(&b[0]);