diff --git a/imperative/python/megengine/module/qat/conv_bn.py b/imperative/python/megengine/module/qat/conv_bn.py
index 3de3509b..4ad64d04 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 1a817c5c..c3c3c330 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"""