1 files changed, 7 insertions, 3 deletions
diff --git a/src/tests/encrypted.rs b/src/tests/encrypted.rs
index aed2662..4705df6 100644
--- a/src/tests/encrypted.rs
+++ b/src/tests/encrypted.rs
@@ -44,7 +44,7 @@ $"#,
     regex::Regex::new(&re).unwrap()
   }
 
-  let mut ncli = Nitrocli::with_model(model);
+  let mut ncli = Nitrocli::make().model(model).build();
   let out = ncli.handle(&["status"])?;
   assert!(make_re(None).is_match(&out), out);
 
@@ -61,7 +61,11 @@ $"#,
 
 #[test_device(pro)]
 fn encrypted_open_on_pro(model: nitrokey::Model) {
-  let res = Nitrocli::with_model(model).handle(&["encrypted", "open"]);
+  let res = Nitrocli::make()
+    .model(model)
+    .build()
+    .handle(&["encrypted", "open"]);
+
   assert_eq!(
     res.unwrap_str_err(),
     "This command is only available on the Nitrokey Storage",
@@ -70,7 +74,7 @@ fn encrypted_open_on_pro(model: nitrokey::Model) {
 
 #[test_device(storage)]
 fn encrypted_open_close(model: nitrokey::Model) -> crate::Result<()> {
-  let mut ncli = Nitrocli::with_model(model);
+  let mut ncli = Nitrocli::make().model(model).build();
   let out = ncli.handle(&["encrypted", "open"])?;
   assert!(out.is_empty());