diff --git a/api/onnx_web/convert/diffusion/diffusion_xl.py b/api/onnx_web/convert/diffusion/diffusion_xl.py
index a92c5856..a7dcf104 100644
--- a/api/onnx_web/convert/diffusion/diffusion_xl.py
+++ b/api/onnx_web/convert/diffusion/diffusion_xl.py
@@ -3,10 +3,8 @@ from os import path
 from typing import Dict, Optional, Tuple
 
 import torch
+from diffusers import StableDiffusionXLPipeline
 from optimum.exporters.onnx import main_export
-from optimum.pipelines.diffusers.pipeline_stable_diffusion_xl import (
-    StableDiffusionXLPipeline,
-)
 
 from ..utils import ConversionContext