diff --git a/imperative/python/megengine/module/qat/conv_bn.py b/imperative/python/megengine/module/qat/conv_bn.py
index 3de3509bfc813969830d86da98af8a4eebe2ebd9..4ad64d04645573e00cec9dbc1f979130a8a73acc 100644
--- a/imperative/python/megengine/module/qat/conv_bn.py
+++ b/imperative/python/megengine/module/qat/conv_bn.py
@@ -153,8 +153,8 @@ class _ConvBnActivation2d(Float._ConvBnActivation2d, QATModule):
             float_module.conv.dilation,
             float_module.conv.groups,
             float_module.conv.bias is not None,
-            float_module.conv.conv_mode.name,
-            float_module.conv.compute_mode.name,
+            float_module.conv.conv_mode,
+            float_module.conv.compute_mode,
         )
         qat_module.conv.weight = float_module.conv.weight
         qat_module.conv.bias = float_module.conv.bias
diff --git a/imperative/python/megengine/module/quantized/linear.py b/imperative/python/megengine/module/quantized/linear.py
index 1a817c5cce7102dd172817b963dfd65570270a23..c3c3c3302681eff78138fb0b453bab4f9ea77cb4 100644
--- a/imperative/python/megengine/module/quantized/linear.py
+++ b/imperative/python/megengine/module/quantized/linear.py
@@ -37,7 +37,6 @@ class Linear(QuantizedModule):
         ret = ret if self.output_dtype is None else ret.astype(self.output_dtype)
         return ret
 
-
     @classmethod
     def from_qat_module(cls, qat_module: QAT.Linear):
         r"""