diff --git a/onnxscript/optimizer/_optimizer.py b/onnxscript/optimizer/_optimizer.py
index 55865e51b6..e017ee205c 100644
--- a/onnxscript/optimizer/_optimizer.py
+++ b/onnxscript/optimizer/_optimizer.py
@@ -4,9 +4,10 @@
 
 import logging
 
+import onnx_ir as ir
 import onnx_ir.passes.common as common_passes
 
-from onnxscript import ir, rewriter
+from onnxscript import rewriter
 from onnxscript.optimizer import _constant_folding
 
 logger = logging.getLogger(__name__)