Update tests

2 files changed, 4 insertions, 4 deletions

diff --git a/tests/knownhosts.rs b/tests/knownhosts.rs
index b1c1f57..aa18865 100644
--- a/tests/knownhosts.rs
+++ b/tests/knownhosts.rs
@@ -1,4 +1,4 @@
-use ssh2::{Session, OpenSSH};
+use ssh2::{Session, KnownHostFileKind};
 
 #[test]
 fn smoke() {
@@ -21,7 +21,7 @@ PW3RcT0eOzQgqlJL3RKrTJvdsjE3JEAvGq3lGHSZXy28G3skua2SmVi\
 ";
     let sess = Session::new().unwrap();
     let mut hosts = sess.known_hosts().unwrap();
-    hosts.read_str(encoded, OpenSSH).unwrap();
+    hosts.read_str(encoded, KnownHostFileKind::OpenSSH).unwrap();
 
     assert_eq!(hosts.iter().count(), 1);
     let host = hosts.iter().next().unwrap().unwrap();
@@ -35,6 +35,6 @@ CZIFImWwoG6mbUoWf9nzpIoaSjB+weqqUUmpaaasXVal72J+UX2B+2R\
 PW3RcT0eOzQgqlJL3RKrTJvdsjE3JEAvGq3lGHSZXy28G3skua2SmVi\
 /w4yCE6gbODqnTWlg7+wC604ydGXA8VJiS5ap43JXiUFFAaQ==");
 
-    assert_eq!(hosts.write_string(&host, OpenSSH).unwrap().as_slice(), encoded);
+    assert_eq!(hosts.write_string(&host, KnownHostFileKind::OpenSSH).unwrap().as_slice(), encoded);
     hosts.remove(host).unwrap();
 }
diff --git a/tests/session.rs b/tests/session.rs
index b13bc95..d8c2c77 100644
--- a/tests/session.rs
+++ b/tests/session.rs
@@ -10,7 +10,7 @@ fn smoke() {
     sess.set_banner("foo").unwrap();
     assert!(sess.is_blocking());
     assert_eq!(sess.timeout(), 0);
-    sess.flag(ssh2::Compress, true).unwrap();
+    sess.flag(ssh2::SessionFlag::Compress, true).unwrap();
     assert!(sess.host_key().is_none());
     sess.method_pref(MethodType::Kex, "diffie-hellman-group14-sha1").unwrap();
     assert!(sess.methods(MethodType::Kex).is_none());