diff --git a/src/encrypt.rs b/src/encrypt.rs
index fb1c633..5d5891f 100644
--- a/src/encrypt.rs
+++ b/src/encrypt.rs
@@ -749,6 +749,7 @@ mod tests {
                     discard_next: false,
                     is_client: false,
                     send_ext_info: true,
+                    strict_kex: false,
                 })
             })
             // and plaintext
diff --git a/src/kex.rs b/src/kex.rs
index 69d5d31..f9e81f0 100644
--- a/src/kex.rs
+++ b/src/kex.rs
@@ -991,8 +991,8 @@ mod tests {
         let serv_init = if let Packet::KexInit(k) = serv_init { k } else { panic!() };
         assert!(ts.next().is_none());
 
-        serv.handle_kexinit(cli_init, false, &serv_conf, &version, &mut ts.sender()).unwrap();
-        cli.handle_kexinit(serv_init, true, &cli_conf, &version, &mut tc.sender()).unwrap();
+        serv.handle_kexinit(cli_init, false, &serv_conf, &version, true, &mut ts.sender()).unwrap();
+        cli.handle_kexinit(serv_init, true, &cli_conf, &version, true, &mut tc.sender()).unwrap();
 
         let cli_dhinit = tc.next().unwrap();
         let cli_dhinit = if let Packet::KexDHInit(k) = cli_dhinit { k } else { panic!() };