diff --git a/api/onnx_web/chain/blend_img2img.py b/api/onnx_web/chain/blend_img2img.py
index 930be4b2..2766bdf1 100644
--- a/api/onnx_web/chain/blend_img2img.py
+++ b/api/onnx_web/chain/blend_img2img.py
@@ -30,7 +30,7 @@ def blend_img2img(
     prompt: str = None,
     **kwargs,
 ) -> Image.Image:
-    logger.info('generating image using img2img', params.prompt)
+    logger.info('generating image using img2img, %s steps: %s', params.steps, params.prompt)
 
     pipe = load_pipeline(OnnxStableDiffusionImg2ImgPipeline,
                             params.model, params.provider, params.scheduler)
diff --git a/api/onnx_web/serve.py b/api/onnx_web/serve.py
index d5c6d802..ac9ab487 100644
--- a/api/onnx_web/serve.py
+++ b/api/onnx_web/serve.py
@@ -588,7 +588,7 @@ def chain():
 
         # TODO: create Upscale from upscale
         if 'upscale' in kwargs:
-            upscale = UpscaleParams(params.model, params.provider)
+            upscale = UpscaleParams(kwargs.get('upscale'), params.provider)
             kwargs['upscale'] = upscale
 
         pipeline.append((callback, stage, kwargs))
diff --git a/api/schema.yaml b/api/schema.yaml
index 079eaa95..98052c82 100644
--- a/api/schema.yaml
+++ b/api/schema.yaml
@@ -16,14 +16,12 @@ $defs:
 
   image_params:
     type: object
-    required: [prompt]
     properties:
       prompt:
         type: string
 
   upscale_params:
     type: object
-    required: [outscale, scale]
     properties:
       outscale:
         type: number
diff --git a/common/pipelines/example.json b/common/pipelines/example.json
index 63a82994..85dc6475 100644
--- a/common/pipelines/example.json
+++ b/common/pipelines/example.json
@@ -19,7 +19,7 @@
       "name": "upscale",
       "type": "upscale-stable-diffusion",
       "params": {
-        "model": "stable-diffusion-x4-upscaler",
+        "upscale": "stable-diffusion-x4-upscaler",
         "prompt": "a magical wizard in a robe fighting a dragon",
         "scale": 4,
         "outscale": 4,