diff --git a/setup.py b/setup.py
index 54b345fc0..e6436e966 100755
--- a/setup.py
+++ b/setup.py
@@ -28,9 +28,9 @@ MOD_NAMES = [
     'spacy.pipeline',
     'spacy.syntax.stateclass',
     'spacy.syntax._state',
-    'spacy.syntax._beam_utils',
     'spacy.tokenizer',
     'spacy.syntax.nn_parser',
+    'spacy.syntax._parser_model',
     'spacy.syntax.nonproj',
     'spacy.syntax.transition_system',
     'spacy.syntax.arc_eager',