diff --git a/olive/passes/onnx/vitis_ai_quantization.py b/olive/passes/onnx/vitis_ai_quantization.py
index 7bf5adf288..afbc95e86f 100644
--- a/olive/passes/onnx/vitis_ai_quantization.py
+++ b/olive/passes/onnx/vitis_ai_quantization.py
@@ -359,7 +359,7 @@ def _run_for_config(
         return model_proto_to_olive_model(onnx_model, output_model_path, config)
 
     def _quant_preprocess(self, model: ONNXModel, output_model_path: str) -> ONNXModel:
-        from olive.passes.onnx.quant_pre_process import quant_pre_process
+        from onnxruntime.quantization.preprocess import quant_pre_process
 
         try:
             quant_pre_process(input_model_path=model.model_path, output_model_path=output_model_path, auto_merge=True)