arm_compute v18.11
diff --git a/tests/validation/reference/Col2Im.cpp b/tests/validation/reference/Col2Im.cpp
index 90e488f..53969d4 100644
--- a/tests/validation/reference/Col2Im.cpp
+++ b/tests/validation/reference/Col2Im.cpp
@@ -40,7 +40,7 @@
SimpleTensor<T> dst{ dst_shape, src.data_type(), 1 };
// Compute reference
- const size_t batches = dst_shape[3];
+ const size_t batches = dst_shape.total_size() / (dst_shape.x() * dst_shape.y() * dst_shape.z());
const size_t src_width = src.shape().x();
const size_t src_height = src.shape().y();