diff --git a/use_case_examples/cifar/cifar_brevitas_training/evaluate_one_example_fhe.py b/use_case_examples/cifar/cifar_brevitas_training/evaluate_one_example_fhe.py
index fff70a89e..afec3d6a8 100644
--- a/use_case_examples/cifar/cifar_brevitas_training/evaluate_one_example_fhe.py
+++ b/use_case_examples/cifar/cifar_brevitas_training/evaluate_one_example_fhe.py
@@ -138,9 +138,8 @@ def wrapper(*args, **kwargs):
     print(f"Quantization of a single input (image) took {quantization_execution_time} seconds")
     print(f"Size of CLEAR input is {q_x_numpy.nbytes} bytes\n")
 
-    p_error = quantized_numpy_module.fhe_circuit.p_error
     expected_quantized_prediction, clear_inference_time = measure_execution_time(
-        partial(quantized_numpy_module.fhe_circuit.simulate, p_error=p_error)
+        partial(quantized_numpy_module.fhe_circuit.simulate)
     )(q_x_numpy)
 
     # Encrypt the input