diff --git a/spacy/ml/staticvectors.py b/spacy/ml/staticvectors.py
index 741007bdb..8dd65833b 100644
--- a/spacy/ml/staticvectors.py
+++ b/spacy/ml/staticvectors.py
@@ -77,7 +77,7 @@ def forward(
         model.inc_grad(
             "W",
             model.ops.gemm(
-                cast(Floats2d, d_output.data), model.ops.as_contig(V[rows]), trans1=True
+                cast(Floats2d, d_output.data), model.ops.as_contig(V), trans1=True
             ),
         )
         return []