diff --git a/Cargo.lock b/Cargo.lock
index 13b78be..334e010 100644
--- a/Cargo.lock
+++ b/Cargo.lock
@@ -118,7 +118,7 @@ dependencies = [
 [[package]]
 name = "concrete-csprng"
 version = "0.4.0"
-source = "git+https://github.com/zama-ai/tfhe-rs.git?rev=tfhe-rs-0.4.1#2f4d00b13a4bce89ca4990365effc5ea6e66bb2d"
+source = "git+https://github.com/zama-ai/tfhe-rs.git?rev=tfhe-rs-0.5.1#bf6b4cc541dd30ca23bd2dbb38f0dc2117f169fe"
 dependencies = [
  "aes",
  "libc",
@@ -219,7 +219,7 @@ checksum = "7fcaabb2fef8c910e7f4c7ce9f67a1283a1715879a7c230ca9d6d1ae31f16d91"
 
 [[package]]
 name = "fhevm-tfhe-cli"
-version = "0.2.3"
+version = "0.2.4"
 dependencies = [
  "bincode",
  "clap",
@@ -515,10 +515,11 @@ checksum = "222a222a5bfe1bba4a77b45ec488a741b3cb8872e5e499451fd7d0129c9c7c3d"
 
 [[package]]
 name = "tfhe"
-version = "0.4.1"
-source = "git+https://github.com/zama-ai/tfhe-rs.git?rev=tfhe-rs-0.4.1#2f4d00b13a4bce89ca4990365effc5ea6e66bb2d"
+version = "0.5.1"
+source = "git+https://github.com/zama-ai/tfhe-rs.git?rev=tfhe-rs-0.5.1#bf6b4cc541dd30ca23bd2dbb38f0dc2117f169fe"
 dependencies = [
  "aligned-vec",
+ "bincode",
  "bytemuck",
  "concrete-csprng",
  "concrete-fft",
diff --git a/Cargo.toml b/Cargo.toml
index 5b3725e..a158df3 100644
--- a/Cargo.toml
+++ b/Cargo.toml
@@ -1,6 +1,6 @@
 [package]
 name = "fhevm-tfhe-cli"
-version = "0.2.3"
+version = "0.2.4"
 authors = ["Zama <hello@zama.ai>"]
 edition = "2021"
 license = "BSD-3-Clause-Clear"
@@ -12,4 +12,4 @@ publish = false
 bincode = "1.3.3"
 clap = { version = "3.1.18", features = ["derive"] }
 serde = "1.0"
-tfhe = { git = "https://github.com/zama-ai/tfhe-rs.git", rev = "tfhe-rs-0.4.1", features = ["boolean", "experimental-force_fft_algo_dif4", "shortint", "integer"] }
+tfhe = { git = "https://github.com/zama-ai/tfhe-rs.git", rev = "tfhe-rs-0.5.1", features = ["boolean", "experimental-force_fft_algo_dif4", "shortint", "integer"] }
diff --git a/README.md b/README.md
index c980d4b..8a0f1da 100644
--- a/README.md
+++ b/README.md
@@ -14,6 +14,7 @@ Tags:
 * v0.2.1: based on tfhe-rs 0.3.1
 * v0.2.2: based on tfhe-rs 0.4.0
 * v0.2.3: based on tfhe-rs 0.4.1
+* v0.2.4: based on tfhe-rs 0.5.1
 
 # Build
 
diff --git a/src/gen_keys.rs b/src/gen_keys.rs
index 35ac2b2..7279803 100644
--- a/src/gen_keys.rs
+++ b/src/gen_keys.rs
@@ -4,14 +4,12 @@
 // All rights reserved.
 
 use tfhe::{
-    generate_keys, shortint::parameters::PARAM_MESSAGE_2_CARRY_2_KS_PBS, 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_KS_PBS, None)
-        .build();
+    let config = ConfigBuilder::with_custom_parameters(PARAM_MESSAGE_2_CARRY_2_KS_PBS, None)
+         .build();
     let (cks, sks) = generate_keys(config);
     let pks = CompactPublicKey::new(&cks);
     (cks, sks, pks)