diff --git a/spacy/tests/parser/test_beam_parse.py b/spacy/tests/parser/test_beam_parse.py
index 59e307bcb..855b2e1e9 100644
--- a/spacy/tests/parser/test_beam_parse.py
+++ b/spacy/tests/parser/test_beam_parse.py
@@ -17,7 +17,7 @@ def test_beam_parse():
     nlp = Language()
     nlp.add_pipe(DependencyParser(nlp.vocab), name='parser')
     nlp.parser.add_label('nsubj')
-    nlp.begin_training()
+    nlp.parser.begin_training([], token_vector_width=8, hidden_width=8)
 
     doc = nlp.make_doc(u'Australia is a country')
     nlp.parser(doc, beam_width=2)