diff --git a/spacy/syntax/parser.pyx b/spacy/syntax/parser.pyx
index b46d42547..3e131e088 100644
--- a/spacy/syntax/parser.pyx
+++ b/spacy/syntax/parser.pyx
@@ -151,7 +151,7 @@ cdef class StepwiseState:
         self.stcls = StateClass.init(doc.c, doc.length)
         self.parser.moves.initialize_state(self.stcls)
         self.eg = Example(self.parser.model.nr_class, CONTEXT_SIZE,
-                          self.parser.model.n_feats, self.parser.model.n_feats)
+                          self.parser.model.nr_feat, self.parser.model.nr_embed)
 
     def __enter__(self):
         return self