diff --git a/mammoth/model_builder.py b/mammoth/model_builder.py
index 36c9e0a3..fb1ad9eb 100644
--- a/mammoth/model_builder.py
+++ b/mammoth/model_builder.py
@@ -16,7 +16,7 @@
     EncoderAdapterLayer,
     DecoderAdapterLayer,
 )
-from mammoth.constants import ModelTask, DefaultTokens
+from mammoth.constants import DefaultTokens
 from mammoth.modules.layer_stack_decoder import LayerStackDecoder
 from mammoth.modules.layer_stack_encoder import LayerStackEncoder
 from mammoth.modules import Embeddings