From 5ce825398a0c55acc8da4b30f8cfe51a54ac0cd1 Mon Sep 17 00:00:00 2001
From: Andrew Poelstra <apoelstra@wpsoftware.net>
Date: Sun, 20 Sep 2015 15:18:53 -0500
Subject: [PATCH] Fix benchmarks

---
 src/ecdh.rs | 2 +-
 src/lib.rs  | 2 +-
 2 files changed, 2 insertions(+), 2 deletions(-)

diff --git a/src/ecdh.rs b/src/ecdh.rs
index 838fe4e..55aeccc 100644
--- a/src/ecdh.rs
+++ b/src/ecdh.rs
@@ -112,6 +112,7 @@ mod benches {
     use rand::{Rng, thread_rng};
     use test::{Bencher, black_box};
 
+    use super::SharedSecret;
     use super::super::{Secp256k1, Message};
 
     #[bench]
@@ -120,7 +121,6 @@ mod benches {
         let (sk, pk) = s.generate_keypair(&mut thread_rng()).unwrap();
 
         let s = Secp256k1::new();
-        let mut r = CounterRng(0);
         bh.iter( || {
             let res = SharedSecret::new(&s, &pk, &sk);
             black_box(res);
diff --git a/src/lib.rs b/src/lib.rs
index 7f7b1b7..794a2c7 100644
--- a/src/lib.rs
+++ b/src/lib.rs
@@ -729,7 +729,7 @@ mod benches {
         thread_rng().fill_bytes(&mut msg);
         let msg = Message::from_slice(&msg).unwrap();
         let (sk, _) = s.generate_keypair(&mut thread_rng()).unwrap();
-        let sig = s.sign(&msg, &sk).unwrap();
+        let sig = s.sign_recoverable(&msg, &sk).unwrap();
 
         bh.iter(|| {
             let res = s.recover(&msg, &sig).unwrap();