diff --git a/examples/notebooks/LeniaEvoJAX.ipynb b/examples/notebooks/LeniaEvoJAX.ipynb
index 7ec812ea..770cac04 100644
--- a/examples/notebooks/LeniaEvoJAX.ipynb
+++ b/examples/notebooks/LeniaEvoJAX.ipynb
@@ -60,7 +60,7 @@
       ],
       "source": [
         "#@title Import\n",
-        "!pip install evojax\n",
+        "!pip install jax==0.4.6 evojax==0.2.15 jaxlib==0.4.6\n",
         "import google.colab\n",
         "google.colab.output.clear()\n",
         "\n",
@@ -3707,4 +3707,4 @@
       "outputs": []
     }
   ]
-}
\ No newline at end of file
+}