diff --git a/how-to-use-azureml/explain-model/azure-integration/gpu-explanation/train-explain-model-gpu-tree-explainer.ipynb b/how-to-use-azureml/explain-model/azure-integration/gpu-explanation/train-explain-model-gpu-tree-explainer.ipynb
index 0d314018..42bbf2fc 100644
--- a/how-to-use-azureml/explain-model/azure-integration/gpu-explanation/train-explain-model-gpu-tree-explainer.ipynb
+++ b/how-to-use-azureml/explain-model/azure-integration/gpu-explanation/train-explain-model-gpu-tree-explainer.ipynb
@@ -241,6 +241,8 @@
         "for dist in list(available_packages):\n",
         "    if dist.key == 'pandas':\n",
         "        pandas_ver = dist.version\n",
+        "    if dist.key == 'numpy':\n",
+        "        numpy_ver = dist.version\n",
         "pandas_dep = 'pandas'\n",
         "numpy_dep = 'numpy'\n",
         "if pandas_ver:\n",
@@ -286,7 +288,7 @@
         "pip uninstall -y xgboost && \\\n",
         "conda install py-xgboost==1.3.3 && \\\n",
         "pip uninstall -y numpy && \\\n",
-        "conda install {numpy_dep} \\\n",
+        "pip install {numpy_dep} \\\n",
         "\"\"\"\n",
         "\n",
         "env.python.user_managed_dependencies = True\n",