diff --git a/src/eva/language/data/datasets/classification/base.py b/src/eva/language/data/datasets/classification/base.py
index ca081ff8..7adaa4db 100644
--- a/src/eva/language/data/datasets/classification/base.py
+++ b/src/eva/language/data/datasets/classification/base.py
@@ -8,7 +8,7 @@
 
 from eva.language.data.datasets.language import LanguageDataset
 
-class LanguageClassification(LanguageDataset[Tuple[str, torch.Tensor]], abc.ABC):
+class TextClassification(LanguageDataset[Tuple[str, torch.Tensor]], abc.ABC):
     """Text classification abstract dataset."""
     
     def __init__(self) -> None:
diff --git a/src/eva/language/data/datasets/classification/pubmedqa.py b/src/eva/language/data/datasets/classification/pubmedqa.py
index 7f185c30..9c07c956 100644
--- a/src/eva/language/data/datasets/classification/pubmedqa.py
+++ b/src/eva/language/data/datasets/classification/pubmedqa.py
@@ -8,7 +8,7 @@
 from eva.language.data.datasets.classification import base
 
 
-class PubMedQA(base.LanguageClassification):
+class PubMedQA(base.TextClassification):
     """Dataset class for PubMedQA question answering task."""
 
     _license: str = "MIT License (https://github.com/pubmedqa/pubmedqa/blob/master/LICENSE)"