diff --git a/src/gen_keys.rs b/src/gen_keys.rs
index 2faec22..35ac2b2 100644
--- a/src/gen_keys.rs
+++ b/src/gen_keys.rs
@@ -4,14 +4,13 @@
 // All rights reserved.
 
 use tfhe::{
-    generate_keys,
-    shortint::parameters::parameters_compact_pk::PARAM_MESSAGE_2_CARRY_2_COMPACT_PK_PBS_KS,
-    ClientKey, CompactPublicKey, ConfigBuilder, ServerKey,
+    generate_keys, shortint::parameters::PARAM_MESSAGE_2_CARRY_2_KS_PBS, ClientKey,
+    CompactPublicKey, ConfigBuilder, ServerKey,
 };
 
 pub fn gen_keys() -> (ClientKey, ServerKey, CompactPublicKey) {
     let config = ConfigBuilder::all_disabled()
-        .enable_custom_integers(PARAM_MESSAGE_2_CARRY_2_COMPACT_PK_PBS_KS, None)
+        .enable_custom_integers(PARAM_MESSAGE_2_CARRY_2_KS_PBS, None)
         .build();
     let (cks, sks) = generate_keys(config);
     let pks = CompactPublicKey::new(&cks);