diff --git a/setup.py b/setup.py
index 1baef0e85..4ee33e255 100644
--- a/setup.py
+++ b/setup.py
@@ -156,6 +156,7 @@ MOD_NAMES = ['spacy.parts_of_speech', 'spacy.strings',
              'spacy.syntax.transition_system',
              'spacy.syntax.arc_eager',
              'spacy.syntax.orig_arc_eager', 
+             'spacy.syntax.tree_arc_eager', 
              'spacy.syntax._parse_features',
              'spacy.gold', 'spacy.orth', 
              'spacy.syntax.ner']