diff --git a/spacy/tests/tokens/test_tokens_api.py b/spacy/tests/tokens/test_tokens_api.py
index bdb0cd3b9..142fa01af 100644
--- a/spacy/tests/tokens/test_tokens_api.py
+++ b/spacy/tests/tokens/test_tokens_api.py
@@ -3,6 +3,7 @@ from __future__ import unicode_literals
 from spacy.tokens import Doc
 from spacy.en import English
 import numpy
+from spacy.attrs import HEAD
 
 import pytest