From 0ce918ff3b1059928b94e1252969b3d771f58062 Mon Sep 17 00:00:00 2001
From: Ben <ben.curtis@zama.ai>
Date: Tue, 6 Feb 2024 14:18:45 +0000
Subject: [PATCH] wrong errortype

---
 estimator/lwe_primal.py | 4 ++--
 1 file changed, 2 insertions(+), 2 deletions(-)

diff --git a/estimator/lwe_primal.py b/estimator/lwe_primal.py
index ce7c186..277ca35 100644
--- a/estimator/lwe_primal.py
+++ b/estimator/lwe_primal.py
@@ -608,8 +608,8 @@ def find_zeta_max(params, red_cost_model):
                         # double it for mitm
                         return 2 * zeta_max
                     zeta_max +=1
-                except ValueError:
-                    pass
+                except NotImplementedError:
+                    return params.n
             return params.n
 
         if zeta is None: