diff --git a/spacy/tests/regression/test_issue3288.py b/spacy/tests/regression/test_issue3288.py
index d17dec971..5379a585a 100644
--- a/spacy/tests/regression/test_issue3288.py
+++ b/spacy/tests/regression/test_issue3288.py
@@ -16,5 +16,5 @@ def test_issue3288(en_vocab):
     heads = [1, 0, -1, 1, 0, 1, -2, -3]
     deps = ["intj", "ROOT", "punct", "advmod", "ROOT", "det", "nsubj", "punct"]
     doc = get_doc(en_vocab, words=words, heads=heads, deps=deps)
-    doc.tensor = numpy.zeros(96, dtype="float32")
+    doc.tensor = numpy.zeros((len(words), 96), dtype="float32")
     displacy.render(doc)