diff --git a/src/core/NEON/kernels/assembly/NEGEMMNativeWrapperKernel.cpp b/src/core/NEON/kernels/assembly/NEGEMMNativeWrapperKernel.cpp
index 38eb40a..ea6a06c 100644
--- a/src/core/NEON/kernels/assembly/NEGEMMNativeWrapperKernel.cpp
+++ b/src/core/NEON/kernels/assembly/NEGEMMNativeWrapperKernel.cpp
@@ -98,8 +98,9 @@
         const unsigned int multi = id.z();
         const unsigned int ymax  = std::min(y0 + strategy::out_height(), m_end);
 
+        // TODO(COMPMID-1424) : Agree on gemm IO layouts
         strat.kernel(a(0, y0, batch, multi), a.stride(Window::DimY),
-                     b(0, 0, 0, multi), b.stride(Window::DimY),
+                     b(0, 0, multi), b.stride(Window::DimY),
                      c(0, y0, batch, multi), c.stride(Window::DimY),
                      _beta, (ymax - y0), _params.N, _params.K);
     });