diff --git a/api/onnx_web/diffusers/run.py b/api/onnx_web/diffusers/run.py
index 04e2f2f4..0477a7f5 100644
--- a/api/onnx_web/diffusers/run.py
+++ b/api/onnx_web/diffusers/run.py
@@ -93,11 +93,12 @@ def add_thumbnail_output(
 
         if metadata.highres is not None:
             metadata.highres = metadata.highres.with_args(
-                outscale=1,
+                scale=1,
             )
 
         if metadata.upscale is not None:
             metadata.upscale = metadata.upscale.with_args(
+                scale=1,
                 outscale=1,
             )