diff --git a/src/eva/vision/data/datasets/segmentation/lits.py b/src/eva/vision/data/datasets/segmentation/lits.py
index 760eadde..6add83b0 100644
--- a/src/eva/vision/data/datasets/segmentation/lits.py
+++ b/src/eva/vision/data/datasets/segmentation/lits.py
@@ -105,8 +105,8 @@ def validate(self) -> None:
         _validators.check_dataset_integrity(
             self,
             length=self._expected_dataset_lengths.get(self._split, 0),
-            n_classes=2,
-            first_and_last_labels=("liver", "tumor"),
+            n_classes=3,
+            first_and_last_labels=("background", "tumor"),
         )
 
     @override