diff --git a/examples_and_demos/kornia_image_points_transforms.ipynb b/examples_and_demos/kornia_image_points_transforms.ipynb
index 21bedaf9..d0cd669d 100644
--- a/examples_and_demos/kornia_image_points_transforms.ipynb
+++ b/examples_and_demos/kornia_image_points_transforms.ipynb
@@ -42,7 +42,7 @@
         "%%capture\n",
         "!pip install kornia\n",
         "!pip install kornia-rs matplotlib\n",
-        "!pip install flax",
+        "!pip install flax==0.10.0 \n",
         "!pip install -q ivy \n",
         "\n",
         "!rm -rf ivy_transpiled_outputs/"