diff --git a/python/hidet/graph/ops/quant/symmetric.py b/python/hidet/graph/ops/quant/symmetric.py
index f377e7853..9885f4ba4 100644
--- a/python/hidet/graph/ops/quant/symmetric.py
+++ b/python/hidet/graph/ops/quant/symmetric.py
@@ -11,7 +11,8 @@
 # limitations under the License.
 from typing import Union, List
 from hidet import ir
-from hidet.ir.type import DataType, int32
+from hidet.ir.type import DataType
+from hidet.ir.dtypes import int32
 from hidet.ir.expr import cast, if_then_else
 from hidet.ir.compute.primitives import TensorNode, compute
 from hidet.ir import primitives as prim