1 files changed, 6 insertions, 8 deletions
diff --git a/src/wireguard/handshake/types.rs b/src/wireguard/handshake/types.rs
index 5f984cc..ed2fcbb 100644
--- a/src/wireguard/handshake/types.rs
+++ b/src/wireguard/handshake/types.rs
@@ -1,10 +1,8 @@
+use super::super::types::KeyPair;
+
 use std::error::Error;
 use std::fmt;
 
-use x25519_dalek::PublicKey;
-
-use super::super::types::KeyPair;
-
 /* Internal types for the noise IKpsk2 implementation */
 
 // config error
@@ -79,10 +77,10 @@ impl Error for HandshakeError {
     }
 }
 
-pub type Output = (
-    Option<PublicKey>, // external identifier associated with peer
-    Option<Vec<u8>>,   // message to send
-    Option<KeyPair>,   // resulting key-pair of successful handshake
+pub type Output<'a, O> = (
+    Option<&'a O>,   // external identifier associated with peer
+    Option<Vec<u8>>, // message to send
+    Option<KeyPair>, // resulting key-pair of successful handshake
 );
 
 // preshared key