diff --git a/examples/demo.ipynb b/examples/demo.ipynb
index 2bc71309..f70cad59 100644
--- a/examples/demo.ipynb
+++ b/examples/demo.ipynb
@@ -88,6 +88,7 @@
       },
       "outputs": [],
       "source": [
+        "!pip install numpy --upgrade\n",
         "!pip install dm-tree\n",
         "!pip install dm-reverb[tensorflow]"
       ]