diff --git a/package-lock.json b/package-lock.json
index 0b3a051..3f04eae 100644
--- a/package-lock.json
+++ b/package-lock.json
@@ -14,9 +14,9 @@
         "ethers": "^6.6.4",
         "libsodium": "^0.7.11",
         "libsodium-wrappers": "^0.7.11",
-        "node-tfhe": "^0.3.0-beta.0",
+        "node-tfhe": "^0.3.1",
         "sha3": "^2.1.4",
-        "tfhe": "^0.3.0-beta.0"
+        "tfhe": "^0.3.1"
       },
       "bin": {
         "fhevm": "bin/fhevm.js"
@@ -4519,9 +4519,9 @@
       "dev": true
     },
     "node_modules/node-tfhe": {
-      "version": "0.3.0-beta.0",
-      "resolved": "https://registry.npmjs.org/node-tfhe/-/node-tfhe-0.3.0-beta.0.tgz",
-      "integrity": "sha512-82t7gbwMV5ASmF5JEwLZK/sbG7SiA9pJ/PCSUhUQmTFJiVNc6PSUPMhpHTqGb3Cnpig2jNQUaKhsonL2+RwyqA=="
+      "version": "0.3.1",
+      "resolved": "https://registry.npmjs.org/node-tfhe/-/node-tfhe-0.3.1.tgz",
+      "integrity": "sha512-re5tBgPMwoSWl6b/pL13MWKnsBUv9nldAXoWkokG1aiUm9P0aOjJoz0J9+zOwV/dB0i9xN/TI//mnqF3uWAOIg=="
     },
     "node_modules/normalize-path": {
       "version": "3.0.0",
@@ -5467,9 +5467,9 @@
       }
     },
     "node_modules/tfhe": {
-      "version": "0.3.0-beta.0",
-      "resolved": "https://registry.npmjs.org/tfhe/-/tfhe-0.3.0-beta.0.tgz",
-      "integrity": "sha512-TjfxwpVkB0BEF1nZOONbztG6G+p+h7oqlXj8+aiopzA1+Mukoc7SnsQvfrrIlNBHdGiSWN0jasDCTv1CS3cc0Q=="
+      "version": "0.3.1",
+      "resolved": "https://registry.npmjs.org/tfhe/-/tfhe-0.3.1.tgz",
+      "integrity": "sha512-PqnsUVChETviwd5ebfJoj9yueUxIKPaSsfSGVmgvpUsUqAlianAx+4UH9hEmjDxgshSRDgm1Fp0/rRgCx2BYog=="
     },
     "node_modules/tmpl": {
       "version": "1.0.5",
@@ -9462,9 +9462,9 @@
       "dev": true
     },
     "node-tfhe": {
-      "version": "0.3.0-beta.0",
-      "resolved": "https://registry.npmjs.org/node-tfhe/-/node-tfhe-0.3.0-beta.0.tgz",
-      "integrity": "sha512-82t7gbwMV5ASmF5JEwLZK/sbG7SiA9pJ/PCSUhUQmTFJiVNc6PSUPMhpHTqGb3Cnpig2jNQUaKhsonL2+RwyqA=="
+      "version": "0.3.1",
+      "resolved": "https://registry.npmjs.org/node-tfhe/-/node-tfhe-0.3.1.tgz",
+      "integrity": "sha512-re5tBgPMwoSWl6b/pL13MWKnsBUv9nldAXoWkokG1aiUm9P0aOjJoz0J9+zOwV/dB0i9xN/TI//mnqF3uWAOIg=="
     },
     "normalize-path": {
       "version": "3.0.0",
@@ -10142,9 +10142,9 @@
       }
     },
     "tfhe": {
-      "version": "0.3.0-beta.0",
-      "resolved": "https://registry.npmjs.org/tfhe/-/tfhe-0.3.0-beta.0.tgz",
-      "integrity": "sha512-TjfxwpVkB0BEF1nZOONbztG6G+p+h7oqlXj8+aiopzA1+Mukoc7SnsQvfrrIlNBHdGiSWN0jasDCTv1CS3cc0Q=="
+      "version": "0.3.1",
+      "resolved": "https://registry.npmjs.org/tfhe/-/tfhe-0.3.1.tgz",
+      "integrity": "sha512-PqnsUVChETviwd5ebfJoj9yueUxIKPaSsfSGVmgvpUsUqAlianAx+4UH9hEmjDxgshSRDgm1Fp0/rRgCx2BYog=="
     },
     "tmpl": {
       "version": "1.0.5",
diff --git a/package.json b/package.json
index 9f4b20c..274bec0 100644
--- a/package.json
+++ b/package.json
@@ -40,9 +40,9 @@
     "ethers": "^6.6.4",
     "libsodium": "^0.7.11",
     "libsodium-wrappers": "^0.7.11",
-    "node-tfhe": "^0.3.0-beta.0",
+    "node-tfhe": "^0.3.1",
     "sha3": "^2.1.4",
-    "tfhe": "^0.3.0-beta.0"
+    "tfhe": "^0.3.1"
   },
   "devDependencies": {
     "@rollup/plugin-alias": "^5.0.0",
diff --git a/src/tfhe.ts b/src/tfhe.ts
index 6a7e40a..899c1be 100644
--- a/src/tfhe.ts
+++ b/src/tfhe.ts
@@ -9,7 +9,7 @@ import { toHexString } from './utils';
 
 export const createTfheKeypair = () => {
   const block_params = new ShortintParameters(
-    ShortintParametersName.PARAM_SMALL_MESSAGE_2_CARRY_2_COMPACT_PK,
+    ShortintParametersName.PARAM_MESSAGE_2_CARRY_2_COMPACT_PK_PBS_KS,
   );
   let config = TfheConfigBuilder.all_disabled()
     .enable_custom_integers(block_params)