diff --git a/crates/model-connector/src/serve.rs b/crates/model-connector/src/serve.rs
index 0919410..57edfa9 100644
--- a/crates/model-connector/src/serve.rs
+++ b/crates/model-connector/src/serve.rs
@@ -887,7 +887,6 @@ mod tests {
             [0xff; 32],
             [0xff; 32],
             0,
-            u64::MAX,
         ))
         .unwrap()
         .len();
diff --git a/crates/model-connector/src/service/lease.rs b/crates/model-connector/src/service/lease.rs
index f677ae0..0e2c37e 100644
--- a/crates/model-connector/src/service/lease.rs
+++ b/crates/model-connector/src/service/lease.rs
@@ -301,9 +301,9 @@ mod tests {
             else {
                 return None;
             };
-            let result = crate::envelope::decrypt_result(&reply, &security(KEY)).unwrap();
+            let result = crate::envelope::decrypt_result(&reply, &security(KEY), &invocation).unwrap();
             match result.disposition {
-                TransportDisposition::Refused { refusal } => Some(refusal),
+                TransportDisposition::Refused(refusal) => Some(refusal),
                 other => panic!("unexpected {other:?}"),
             }
         };