diff --git a/combo/modules/lemma.py b/combo/modules/lemma.py
index c4c0495da97fe18f369b108312eb8ebc454293d4..9ef9e5e28f5e7a5f67face2448fcf664a20ba40c 100644
--- a/combo/modules/lemma.py
+++ b/combo/modules/lemma.py
@@ -9,11 +9,11 @@ from combo.config import Registry
 from combo.config.from_parameters import register_arguments
 from combo.modules import dilated_cnn
 from combo.nn import base
-from combo.nn.base import Predictor
 from combo.nn.activations import Activation
 from combo.nn.utils import masked_cross_entropy
 from combo.utils import ConfigurationError
 from combo.models.base import TimeDistributed
+from combo.predictors import Predictor
 
 
 @Registry.register('combo_lemma_predictor_from_vocab')