diff --git a/setup.py b/setup.py
index d16615f5f..864a4036a 100755
--- a/setup.py
+++ b/setup.py
@@ -35,13 +35,14 @@ MOD_NAMES = [
     "spacy.syntax.stateclass",
     "spacy.syntax._state",
     "spacy.tokenizer",
+    "spacy.syntax.gold_parse",
     "spacy.syntax.nn_parser",
     "spacy.syntax._parser_model",
     "spacy.syntax._beam_utils",
     "spacy.syntax.nonproj",
     "spacy.syntax.transition_system",
     "spacy.syntax.arc_eager",
-    "spacy.gold",
+    "spacy.gold.gold_io",
     "spacy.tokens.doc",
     "spacy.tokens.span",
     "spacy.tokens.token",
diff --git a/spacy/syntax/_beam_utils.pyx b/spacy/syntax/_beam_utils.pyx
index 03702e54e..46bff1af9 100644
--- a/spacy/syntax/_beam_utils.pyx
+++ b/spacy/syntax/_beam_utils.pyx
@@ -9,7 +9,7 @@ import numpy
 
 from ..typedefs cimport hash_t, class_t
 from .transition_system cimport TransitionSystem, Transition
-from ..gold cimport GoldParse
+from .gold_parse cimport GoldParse
 from .stateclass cimport StateC, StateClass
 
 from ..errors import Errors