diff --git a/datasets/flwr_datasets/partitioner/semantic_partitioner.py b/datasets/flwr_datasets/partitioner/semantic_partitioner.py
index c4aa05eddea6..f9fca53b0a3e 100644
--- a/datasets/flwr_datasets/partitioner/semantic_partitioner.py
+++ b/datasets/flwr_datasets/partitioner/semantic_partitioner.py
@@ -98,7 +98,7 @@ class SemanticPartitioner(Partitioner):
     Examples
     --------
     >>> from flwr_datasets import FederatedDataset
-    >>> from flwr_datasets import SemanticPartitioner
+    >>> from flwr_datasets.partitioner import SemanticPartitioner
     >>> partitioner = SemanticPartitioner(
     >>>     num_partitions=10,
     >>>     partition_by="label",