diff --git a/trojanvision/configs/defense/neuron_inspect.yml b/trojanvision/configs/defense/neuron_inspect.yml
index 667334eb..0eac2ec5 100644
--- a/trojanvision/configs/defense/neuron_inspect.yml
+++ b/trojanvision/configs/defense/neuron_inspect.yml
@@ -1,6 +1,6 @@
 lambd_sp: 1.e-6
 lambd_sm: 1.e-6
-lambd_pe: 1
+lambd_pe: 1.
 thre: 1.e-5
 
 sample_ratio: 0.1
\ No newline at end of file
diff --git a/trojanvision/defenses/backdoor/model_inspection/neuron_inspect.py b/trojanvision/defenses/backdoor/model_inspection/neuron_inspect.py
index 39dacc73..f74b8c5c 100644
--- a/trojanvision/defenses/backdoor/model_inspection/neuron_inspect.py
+++ b/trojanvision/defenses/backdoor/model_inspection/neuron_inspect.py
@@ -26,8 +26,8 @@ def add_argument(cls, group: argparse._ArgumentGroup):
         group.add_argument('--sample_ratio', type=float, help='sample ratio from the full training data')
         return group
 
-    def __init__(self, lambd_sp: float = 1e-5, lambd_sm: float = 1e-5, lambd_pe: float = 1,
-                 thre: float = 0, sample_ratio: float = 0.1, **kwargs):
+    def __init__(self, lambd_sp: float = 1e-5, lambd_sm: float = 1e-5, lambd_pe: float = 1.,
+                 thre: float = 0., sample_ratio: float = 0.1, **kwargs):
         super().__init__(**kwargs)
         self.param_list['neuron_inspect'] = ['lambd_sp', 'lambd_sm', 'lambd_pe', 'thre', 'sample_ratio']