diff --git a/spacy/syntax/nonproj.pyx b/spacy/syntax/nonproj.pyx
index a91176f44..ee3219392 100644
--- a/spacy/syntax/nonproj.pyx
+++ b/spacy/syntax/nonproj.pyx
@@ -7,7 +7,7 @@ from copy import copy
 
 from ..tokens.doc cimport Doc, set_children_from_heads
 
-from ..gold import Example
+from ..gold import Example, TokenAnnotation
 from ..errors import Errors