diff --git a/fooocusapi/worker.py b/fooocusapi/worker.py
index d5b967a..60a3f67 100644
--- a/fooocusapi/worker.py
+++ b/fooocusapi/worker.py
@@ -409,7 +409,7 @@ def yield_result(_, images, tasks, extension='png'):
                             inpaint_mask_image_upload = resample_image(inpaint_mask_image_upload, width=W, height=H)
                             inpaint_mask_image_upload = np.mean(inpaint_mask_image_upload, axis=2)
                             inpaint_mask_image_upload = (inpaint_mask_image_upload > 127).astype(np.uint8) * 255
-                            inpaint_mask = np.maximum(np.zeros(shape=(W, H), dtype=np.uint8), inpaint_mask_image_upload)
+                            inpaint_mask = np.maximum(np.zeros(shape=(H, W), dtype=np.uint8), inpaint_mask_image_upload)
 
                 if int(inpaint_erode_or_dilate) != 0:
                     inpaint_mask = erode_or_dilate(inpaint_mask, inpaint_erode_or_dilate)