diff --git a/api/onnx_web/chain/highres.py b/api/onnx_web/chain/highres.py
index 2318223d..f48f7c40 100644
--- a/api/onnx_web/chain/highres.py
+++ b/api/onnx_web/chain/highres.py
@@ -72,7 +72,7 @@ def stage_highres(
         EditMetadataStage(),
         stage.with_args(outscale=1, tile_size=SizeChart.max),
         highres=highres,
-        upscale=UpscaleParams(upscale.upscale_model, upscale=False),
+        upscale=upscale,
         replace_params=params,
     )