diff --git a/spacy/ml/models/parser.py b/spacy/ml/models/parser.py
index 0b0676b7f..0d3baf0ec 100644
--- a/spacy/ml/models/parser.py
+++ b/spacy/ml/models/parser.py
@@ -1,5 +1,6 @@
-from typing import Optional
+from typing import Optional, List
 from thinc.api import Model, chain, list2array, Linear, zero_init, use_ops
+from thinc.types import Floats2d
 
 from ...util import registry
 from .._precomputable_affine import PrecomputableAffine