diff --git a/src/cpu/kernels/CpuGemmLowpOffsetContributionOutputStageKernel.cpp b/src/cpu/kernels/CpuGemmLowpOffsetContributionOutputStageKernel.cpp
index c69af55..190487e 100644
--- a/src/cpu/kernels/CpuGemmLowpOffsetContributionOutputStageKernel.cpp
+++ b/src/cpu/kernels/CpuGemmLowpOffsetContributionOutputStageKernel.cpp
@@ -839,7 +839,7 @@
         }
 
         // Check Tensor Rank of vector_sum_row
-        ARM_COMPUTE_RETURN_ERROR_ON(vector_sum_row->num_dimensions() > 2);
+        ARM_COMPUTE_RETURN_ERROR_ON(vector_sum_row->num_dimensions() > 3);
     }
 
     if(output->total_size() != 0)