diff --git a/src/core/Helpers.cpp b/src/core/Helpers.cpp
index 5c7200b..6701ff4 100644
--- a/src/core/Helpers.cpp
+++ b/src/core/Helpers.cpp
@@ -246,7 +246,7 @@
     }
 
     // Setup output valid region
-    ValidRegion valid_region{ Coordinates(), dst_shape, src_info.tensor_shape().num_dimensions() };
+    ValidRegion valid_region{ Coordinates(), dst_shape, dst_shape.num_dimensions() };
 
     valid_region.anchor.set(idx_width, std::max(0, valid_start_out_x));
     valid_region.anchor.set(idx_height, std::max(0, valid_start_out_y));