diff --git a/examples/text_summarization/abstractive_summarization_unilm_cnndm.ipynb b/examples/text_summarization/abstractive_summarization_unilm_cnndm.ipynb
index ee6c51c..9927893 100644
--- a/examples/text_summarization/abstractive_summarization_unilm_cnndm.ipynb
+++ b/examples/text_summarization/abstractive_summarization_unilm_cnndm.ipynb
@@ -65,7 +65,7 @@
     "import time\n",
     "\n",
     "from utils_nlp.dataset.cnndm import CNNDMSummarizationDatasetOrg\n",
-    "from utils_nlp.models.abstractive_summarization_seq2seq import S2SAbsSumProcessor, S2SAbstractiveSummarizer\n",
+    "from utils_nlp.models.transformers.abstractive_summarization_seq2seq import S2SAbsSumProcessor, S2SAbstractiveSummarizer\n",
     "from utils_nlp.eval import compute_rouge_python\n",
     "\n",
     "start_time = time.time()"
diff --git a/examples/text_summarization/abstractive_summarization_unilm_cnndm.py b/examples/text_summarization/abstractive_summarization_unilm_cnndm.py
index 1152468..0cac938 100644
--- a/examples/text_summarization/abstractive_summarization_unilm_cnndm.py
+++ b/examples/text_summarization/abstractive_summarization_unilm_cnndm.py
@@ -4,7 +4,7 @@ import jsonlines
 
 import torch
 
-from utils_nlp.models.abstractive_summarization_seq2seq import (
+from utils_nlp.models.transformers.abstractive_summarization_seq2seq import (
      S2SAbsSumProcessor, 
      S2SAbstractiveSummarizer
 )
diff --git a/tests/unit/test_abstractive_summarization_seq2seq.py b/tests/unit/test_abstractive_summarization_seq2seq.py
index 81847ca..cb0bc6c 100644
--- a/tests/unit/test_abstractive_summarization_seq2seq.py
+++ b/tests/unit/test_abstractive_summarization_seq2seq.py
@@ -4,7 +4,7 @@
 import os
 import pytest
 
-from utils_nlp.models.abstractive_summarization_seq2seq import (
+from utils_nlp.models.transformers.abstractive_summarization_seq2seq import (
     S2SAbsSumProcessor, 
     S2SAbstractiveSummarizer, 
     S2SConfig