diff --git a/spacy/pipeline/edit_tree_lemmatizer.py b/spacy/pipeline/edit_tree_lemmatizer.py
index 7f6367c75..76b0e0bc9 100644
--- a/spacy/pipeline/edit_tree_lemmatizer.py
+++ b/spacy/pipeline/edit_tree_lemmatizer.py
@@ -149,7 +149,7 @@ class EditTreeLemmatizer(TrainablePipe):
             # Handle cases where there are no tokens in any docs.
             n_labels = len(self.cfg["labels"])
             guesses: List[Ints2d] = [
-                self.model.ops.alloc2i(0, n_labels, dtype="i") for _ in docs
+                self.model.ops.alloc2i(0, n_labels) for _ in docs
             ]
             assert len(guesses) == n_docs
             return guesses