diff --git a/examples/PyMPDATA_examples/advection_diffusion_2d/advection-diffusion-2d.ipynb b/examples/PyMPDATA_examples/advection_diffusion_2d/advection-diffusion-2d.ipynb
index c843a777..fbc5cf35 100644
--- a/examples/PyMPDATA_examples/advection_diffusion_2d/advection-diffusion-2d.ipynb
+++ b/examples/PyMPDATA_examples/advection_diffusion_2d/advection-diffusion-2d.ipynb
@@ -10,6 +10,24 @@
     "[![Open In Colab](https://colab.research.google.com/assets/colab-badge.svg)](https://colab.research.google.com/github/open-atmos/PyMPDATA/blob/main/examples/PyMPDATA_examples/advection_diffusion_2d/advection-diffusion-2d.ipynb)"
    ]
   },
+  {
+   "cell_type": "code",
+   "id": "e333666d",
+   "execution_count": null,
+   "metadata": {
+    "pycharm": {
+     "name": "#%%\n"
+    }
+   },
+   "outputs": [],
+   "source": [
+    "import sys\n",
+    "if 'google.colab' in sys.modules:\n",
+    "    !pip --quiet install open-atmos-jupyter-utils\n",
+    "    from open_atmos_jupyter_utils import pip_install_on_colab\n",
+    "    pip_install_on_colab('PyMPDATA-examples')"
+   ]
+  },
   {
    "cell_type": "code",
    "id": "43d2893d-f472-43ac-ad5b-bf342a3783b9",