diff --git a/api/onnx_web/diffusers/utils.py b/api/onnx_web/diffusers/utils.py
index 19f3130d..ff39d371 100644
--- a/api/onnx_web/diffusers/utils.py
+++ b/api/onnx_web/diffusers/utils.py
@@ -296,9 +296,9 @@ def expand_latents(
 
 def resize_latent_shape(
     latents: np.ndarray,
-    size: Size,
+    size: Tuple[int, int],
 ) -> Tuple[int, int, int, int]:
-    return (latents.shape[0], latents.shape[1], size.height, size.width)
+    return (latents.shape[0], latents.shape[1], *size)
 
 
 def get_tile_latents(