diff --git a/mindarmour/diff_privacy/train/model.py b/mindarmour/diff_privacy/train/model.py
index 17c49275fc0f8a391a54e850383a9a07d88402ff..1ad16754084c8968f245b01682fc697fa8ab9115 100644
--- a/mindarmour/diff_privacy/train/model.py
+++ b/mindarmour/diff_privacy/train/model.py
@@ -114,8 +114,8 @@ class DPModel(Model):
             self._micro_batches = check_int_positive('micro_batches', micro_batches)
         else:
             self._micro_batches = None
-        float_norm_clip = check_param_type('l2_norm_clip', norm_clip, float)
-        self._norm_clip = check_value_positive('l2_norm_clip', float_norm_clip)
+        norm_clip = check_param_type('norm_clip', norm_clip, float)
+        self._norm_clip = check_value_positive('norm_clip', norm_clip)
         if isinstance(dp_mech, mechanisms.Mechanisms):
             self._dp_mech = dp_mech
         else: