diff --git a/combo/predict.py b/combo/predict.py
index 803928fff4bb531a626754a89982b6910a374c9c..fcc8fff031b4d55ef10878d0f6b5766067146207 100644
--- a/combo/predict.py
+++ b/combo/predict.py
@@ -10,6 +10,7 @@ from allennlp.data import tokenizers
 from allennlp.predictors import predictor
 from overrides import overrides
 
+from combo import data
 from combo.data import sentence2conllu, tokens2conllu, conllu2sentence
 from combo.utils import download, graph, lambo