diff --git a/how-to-use-azureml/work-with-data/datasets-tutorial/pipeline-with-datasets/pipeline-for-image-classification.ipynb b/how-to-use-azureml/work-with-data/datasets-tutorial/pipeline-with-datasets/pipeline-for-image-classification.ipynb
index 59928c8d..648dfce4 100644
--- a/how-to-use-azureml/work-with-data/datasets-tutorial/pipeline-with-datasets/pipeline-for-image-classification.ipynb
+++ b/how-to-use-azureml/work-with-data/datasets-tutorial/pipeline-with-datasets/pipeline-for-image-classification.ipynb
@@ -272,7 +272,8 @@
         "dependencies:\n",
         "- python=3.6.2\n",
         "- pip:\n",
-        "  - azureml-defaults\n",
+        "  - azureml-core\n",
+        "  - azureml-dataset-runtime\n",
         "  - keras==2.4.3\n",
         "  - tensorflow==2.4.3\n",
         "  - numpy\n",