diff --git a/docs/source/data/partitioned_and_incremental_datasets.md b/docs/source/data/partitioned_and_incremental_datasets.md
index d7a77bd03b..ebb56e2c15 100644
--- a/docs/source/data/partitioned_and_incremental_datasets.md
+++ b/docs/source/data/partitioned_and_incremental_datasets.md
@@ -38,7 +38,7 @@ Like any other dataset, `PartitionedDataset` can also be instantiated programmat
 
 ```python
 from kedro_datasets.pandas import CSVDataset
-from kedro.io import PartitionedDataset
+from kedro_datasets.partitions import PartitionedDataset
 
 my_credentials = {...}  # credentials dictionary