f

zama-ai · Oct 31, 2023 · 7df4a9d · 7df4a9d
1 parent a277cfb
commit 7df4a9d
Showing 1 changed file with 4 additions and 2 deletions.
diff --git a/tfhe/src/shortint/server_key/tests/noise_level.rs b/tfhe/src/shortint/server_key/tests/noise_level.rs
@@ -255,10 +255,12 @@ fn test_ct_scalar_op_assign_noise_level_propagation(sk: &ServerKey, ct: &Ciphert
 
 #[test]
 fn test_noise_level_propagation() {
-    let keys = KEY_CACHE.get_from_param(PARAM_MESSAGE_2_CARRY_2_KS_PBS);
+    let params = PARAM_MESSAGE_2_CARRY_2_KS_PBS;
+
+    let keys = KEY_CACHE.get_from_param(params);
     let (ck, sk) = (keys.client_key(), keys.server_key());
 
-    let modulus: u64 = 4;
+    let modulus: u64 = params.message_modulus.0 as u64;
 
     for _ in 0..10 {
         let trivial0 = sk.create_trivial(0);