diff --git a/datasets/flwr_datasets/preprocessor/divider.py b/datasets/flwr_datasets/preprocessor/divider.py
index c2ee8e1a1a6a..9d7570de4cea 100644
--- a/datasets/flwr_datasets/preprocessor/divider.py
+++ b/datasets/flwr_datasets/preprocessor/divider.py
@@ -66,14 +66,14 @@ class Divider:
 
     >>> # Assuming there is a dataset_dict of type `DatasetDict`
     >>> # dataset_dict is {"train": train-data, "test": test-data}
-    >>> resplitter = Divider(
+    >>> divider = Divider(
     >>>     divide_config={
     >>>         "train": 0.8,
     >>>         "valid": 0.2,
     >>>     }
     >>>     divide_split="train",
     >>> )
-    >>> new_dataset_dict = resplitter(dataset_dict)
+    >>> new_dataset_dict = divider(dataset_dict)
     >>> # new_dataset_dict is
     >>> # {"train": 80% of train, "valid": 20% of train, "test": test-data}
 
@@ -83,7 +83,7 @@ class Divider:
 
     >>> # Assuming there is a dataset_dict of type `DatasetDict`
     >>> # dataset_dict is {"train": train-data, "test": test-data}
-    >>> resplitter = Divider(
+    >>> divider = Divider(
     >>>     divide_config={
     >>>         "train": {
     >>>             "train": 0.8,
@@ -92,7 +92,7 @@ class Divider:
     >>>         "test": {"test-a": 0.4, "test-b": 0.6 }
     >>>     }
     >>> )
-    >>> new_dataset_dict = resplitter(dataset_dict)
+    >>> new_dataset_dict = divider(dataset_dict)
     >>> # new_dataset_dict is
     >>> # {"train": 80% of train, "valid": 20% of train,
     >>> # "test-a": 40% of test, "test-b": 60% of test}
diff --git a/datasets/flwr_datasets/preprocessor/divider_test.py b/datasets/flwr_datasets/preprocessor/divider_test.py
index 586e73f241a6..ed282fbc18be 100644
--- a/datasets/flwr_datasets/preprocessor/divider_test.py
+++ b/datasets/flwr_datasets/preprocessor/divider_test.py
@@ -80,7 +80,7 @@
         ),
     ],
 )
-class TestDivideResplitter(unittest.TestCase):
+class TestDivider(unittest.TestCase):
     """Divider tests."""
 
     divide_config: Union[
@@ -125,7 +125,7 @@ def test_resplitting_correct_new_split_sizes(self) -> None:
 
 
 class TestDividerIncorrectUseCases(unittest.TestCase):
-    """Resplitter tests."""
+    """Divider tests."""
 
     def setUp(self) -> None:
         """Set up the dataset with 3 splits for tests."""