diff --git a/src/languedoc/train.py b/src/languedoc/train.py
--- a/src/languedoc/train.py
+++ b/src/languedoc/train.py
@@ -4,7 +4,7 @@ import itertools
 import json
 import gzip
 
-from predict import preprocess, identify, extract_ngram_freqs, rank_ngram_freqs, Sample
+from .predict import preprocess, identify, extract_ngram_freqs, rank_ngram_freqs, Sample
 
 random.seed(19181028)