diff --git a/spacy/cli/train.py b/spacy/cli/train.py
index 3420c96fa..6a1d74934 100644
--- a/spacy/cli/train.py
+++ b/spacy/cli/train.py
@@ -12,7 +12,7 @@ import thinc.schedules
 from thinc.api import Model, use_pytorch_for_gpu_memory
 import random
 
-from ..gold.corpus_docbin import Corpus
+from ..gold import Corpus
 from ..lookups import Lookups
 from .. import util
 from ..errors import Errors