diff --git a/api/onnx_web/chain/upscale_outpaint.py b/api/onnx_web/chain/upscale_outpaint.py
index 6d8d9587..0f552bd5 100644
--- a/api/onnx_web/chain/upscale_outpaint.py
+++ b/api/onnx_web/chain/upscale_outpaint.py
@@ -105,7 +105,7 @@ def upscale_outpaint(
                 guidance_scale=params.cfg,
                 height=size.height,
                 latents=latents,
-                mask=tile_mask,
+                mask_image=tile_mask,
                 negative_prompt=params.negative_prompt,
                 num_inference_steps=params.steps,
                 width=size.width,
diff --git a/api/scripts/test-memory-leak.sh b/api/scripts/test-memory-leak.sh
new file mode 100644
index 00000000..42d2cdba
--- /dev/null
+++ b/api/scripts/test-memory-leak.sh
@@ -0,0 +1,18 @@
+test_host="${1:-127.0.0.1}"
+test_images=0
+
+while true;
+do
+  curl "http://${test_host}:5000/api/txt2img?"\
+    'cfg=16.00&steps=35&scheduler=deis-multi&seed=-1&'\
+    'prompt=an+astronaut+eating+a+hamburger&negativePrompt=&'\
+    'model=stable-diffusion-v1-5&platform=any&'\
+    'upscaling=upscaling-real-esrgan-x2-plus&correction=correction-codeformer&'\
+    'lpw=false&width=512&height=512&upscaleOrder=correction-both' \
+    -X 'POST' \
+    --compressed \
+    --insecure || break;
+  ((test_images++));
+  echo "waiting after $test_images";
+  sleep 30;
+done