diff --git a/src/correlation.rs b/src/correlation.rs
index 8972f17..7b67541 100644
--- a/src/correlation.rs
+++ b/src/correlation.rs
@@ -970,7 +970,7 @@ mod gpu {
         ) -> Result<(), Box<dyn error::Error>> {
             let max_width = img1.width().max(img2.width());
             let max_height = img1.height().max(img2.height());
-            let max_shape = (max_height, max_width);
+            let max_shape = (max_width, max_height);
             let img1_shape = (img1.width(), img1.height());
             let out_shape = match dir {
                 CorrelationDirection::Forward => self.img1_shape,