diff --git a/docs/apache-airflow/authoring-and-scheduling/datasets.rst b/docs/apache-airflow/authoring-and-scheduling/datasets.rst
index 8fbff61a836a2..605635a4e1d08 100644
--- a/docs/apache-airflow/authoring-and-scheduling/datasets.rst
+++ b/docs/apache-airflow/authoring-and-scheduling/datasets.rst
@@ -27,7 +27,7 @@ In addition to scheduling DAGs based upon time, they can also be scheduled based
 
 .. code-block:: python
 
-    from airflow import Dataset
+    from airflow.datasets import Dataset
 
     with DAG(...):
         MyOperator(
@@ -57,7 +57,7 @@ A dataset is defined by a Uniform Resource Identifier (URI):
 
 .. code-block:: python
 
-    from airflow import Dataset
+    from airflow.datasets import Dataset
 
     example_dataset = Dataset("s3://dataset-bucket/example.csv")