diff --git a/spacy/ml/featureextractor.py b/spacy/ml/featureextractor.py
index 3d189008a..ed2918f02 100644
--- a/spacy/ml/featureextractor.py
+++ b/spacy/ml/featureextractor.py
@@ -1,7 +1,8 @@
 from typing import List, Union, Callable, Tuple
-from thinc.types import Ints2d, Doc
+from thinc.types import Ints2d
 from thinc.api import Model, registry
 
+from ..tokens import Doc
 
 
 @registry.layers("spacy.FeatureExtractor.v1")