diff --git a/tests/unit/test_models_transformers_question_answering.py b/tests/unit/test_models_transformers_question_answering.py
index bbd3d71..151b898 100644
--- a/tests/unit/test_models_transformers_question_answering.py
+++ b/tests/unit/test_models_transformers_question_answering.py
@@ -3,7 +3,7 @@
 
 import pytest
 import os
-from utils_nlp.dataset.pytorch import QADataset
+from utils_nlp.models.transformers.datasets import QADataset
 from utils_nlp.models.transformers.question_answering import (
     QAProcessor,
     AnswerExtractor,