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 0637c41c..59928c8d 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
@@ -273,8 +273,8 @@
         "- python=3.6.2\n",
         "- pip:\n",
         "  - azureml-defaults\n",
-        "  - keras\n",
-        "  - tensorflow<=2.4.*\n",
+        "  - keras==2.4.3\n",
+        "  - tensorflow==2.4.3\n",
         "  - numpy\n",
         "  - scikit-learn\n",
         "  - pandas\n",