diff --git a/combo/combo_model.py b/combo/combo_model.py
index d87d514adeddb223bec672e43d664d347f97cf82..162a3b18522daeae810cbc8ecf930fc0ad1b4972 100644
--- a/combo/combo_model.py
+++ b/combo/combo_model.py
@@ -24,7 +24,7 @@ from combo.nn.utils import get_text_field_mask
 from combo.predictors import Predictor
 from combo.utils import metrics
 from combo.utils import ConfigurationError
-from modules.seq2seq_encoders.transformer_encoder import TransformerEncoder
+from combo.modules.seq2seq_encoders.transformer_encoder import TransformerEncoder
 
 
 @Registry.register("semantic_multitask")
diff --git a/combo/modules/seq2seq_encoders/transformer_encoder.py b/combo/modules/seq2seq_encoders/transformer_encoder.py
index a49100def51df028bd8427ea65e0ab8bd818905a..3b126fac38009a82cfbe3efa259c7cd2c6a4a2af 100644
--- a/combo/modules/seq2seq_encoders/transformer_encoder.py
+++ b/combo/modules/seq2seq_encoders/transformer_encoder.py
@@ -1,18 +1,11 @@
 from typing import Optional
 
-from overrides import overrides
 import torch
 from torch import nn
 
 from combo.modules.encoder import _EncoderBase
 from combo.config.from_parameters import FromParameters, register_arguments
-
-# from modules.seq2seq_encoders.seq2seq_encoder import Seq2SeqEncoder
-from nn.utils import add_positional_features
-
-
-# from allennlp.modules.seq2seq_encoders.seq2seq_encoder import Seq2SeqEncoder
-# from allennlp.nn.util import add_positional_features
+from combo.nn.utils import add_positional_features
 
 
 class TransformerEncoder(_EncoderBase, FromParameters):