diff --git a/combo/modules/token_embedders/token_embedder.py b/combo/modules/token_embedders/token_embedder.py
index 0a60351b3c3454f51a137646b09be6192ced568e..de21b9f3eb58d820e93c58cf176b9981b18a3ba1 100644
--- a/combo/modules/token_embedders/token_embedder.py
+++ b/combo/modules/token_embedders/token_embedder.py
@@ -12,7 +12,7 @@ from combo.data import Vocabulary
 from combo.nn.utils import tiny_value_of_dtype, uncombine_initial_dims, combine_initial_dims
 from combo.modules.module import Module
 from combo.utils import ConfigurationError
-from models.time_distributed import TimeDistributed
+from combo.models.time_distributed import TimeDistributed
 
 
 class TokenEmbedder(Module, FromParameters):