diff --git a/rh_utils.py b/rh_utils.py
index 73e4cd2..49721fa 100644
--- a/rh_utils.py
+++ b/rh_utils.py
@@ -48,11 +48,11 @@ def decode_latents(flux_dir, latents):
         torch_dtype=torch.bfloat16,
     ).to("cuda")
 
-    latents = pipeline._unpack_latents(latents, g_width, g_height, pipeline.vae_scale_factor)
+    latents = pipeline._unpack_latents(latents, g_height, g_width, pipeline.vae_scale_factor)
     latents = (
         latents / pipeline.vae.config.scaling_factor
     ) + pipeline.vae.config.shift_factor
     image = pipeline.vae.decode(latents, return_dict=False)[0]
     image = pipeline.image_processor.postprocess(image, output_type="pil")
 
-    return FluxPipelineOutput(images=image)
\ No newline at end of file
+    return FluxPipelineOutput(images=image)