diff --git a/api/onnx_web/diffusion/load.py b/api/onnx_web/diffusion/load.py
index 23f5d6cf..64e2908e 100644
--- a/api/onnx_web/diffusion/load.py
+++ b/api/onnx_web/diffusion/load.py
@@ -1,5 +1,5 @@
 from logging import getLogger
-from typing import Any, Tuple
+from typing import Any, Optional, Tuple
 
 import numpy as np
 from diffusers import DiffusionPipeline
@@ -9,9 +9,15 @@ from ..utils import run_gc
 
 logger = getLogger(__name__)
 
-last_pipeline_instance = None
-last_pipeline_options = (None, None, None)
-last_pipeline_scheduler = None
+last_pipeline_instance: Any = None
+last_pipeline_options: Tuple[
+    Optional[DiffusionPipeline],
+    Optional[str],
+    Optional[str],
+    Optional[str],
+    Optional[bool],
+] = (None, None, None, None, None)
+last_pipeline_scheduler: Any = None
 
 latent_channels = 4
 latent_factor = 8
diff --git a/api/onnx_web/diffusion/run.py b/api/onnx_web/diffusion/run.py
index 5f4636e2..e8924b94 100644
--- a/api/onnx_web/diffusion/run.py
+++ b/api/onnx_web/diffusion/run.py
@@ -111,8 +111,8 @@ def run_img2img_pipeline(
     else:
         rng = np.random.RandomState(params.seed)
         result = pipe(
-            source_image,
             params.prompt,
+            source_image,
             generator=rng,
             guidance_scale=params.cfg,
             negative_prompt=params.negative_prompt,