diff --git a/setup.py b/setup.py
index 397091403..837d7dd05 100644
--- a/setup.py
+++ b/setup.py
@@ -50,8 +50,11 @@ exts = [
     Extension("spacy.lexeme", ["spacy/lexeme.pyx"], language="c++", include_dirs=includes),
     Extension("spacy.en", ["spacy/en.pyx"], language="c++", include_dirs=includes),
     Extension("spacy.tokens", ["spacy/tokens.pyx"], language="c++", include_dirs=includes),
-    Extension("spacy.pos", ["spacy/pos.pyx"], language="c++", include_dirs=includes),
+    Extension("spacy.tagger", ["spacy/tagger.pyx"], language="c++", include_dirs=includes),
+    Extension("spacy.context", ["spacy/context.pyx"], language="c++", include_dirs=includes),
     Extension("spacy.utf8string", ["spacy/utf8string.pyx"], language="c++", include_dirs=includes),
+    Extension("spacy.pos_feats", ["spacy/pos_feats.pyx"], language="c++", include_dirs=includes),
+    Extension("spacy.ner_feats", ["spacy/ner_feats.pyx"], language="c++", include_dirs=includes)
 ]