diff --git a/docs/getting_started.ipynb b/docs/getting_started.ipynb
index c05109314e..13858675b7 100644
--- a/docs/getting_started.ipynb
+++ b/docs/getting_started.ipynb
@@ -41,7 +41,7 @@
    },
    "outputs": [],
    "source": [
-    "!pip install -q flax"
+    "!pip install -q flax>=0.7.5"
    ]
   },
   {
diff --git a/docs/getting_started.md b/docs/getting_started.md
index c8edb350ff..e14ef4c97e 100644
--- a/docs/getting_started.md
+++ b/docs/getting_started.md
@@ -30,7 +30,7 @@ the network for image classification on the MNIST dataset.
 :id: bb81587e
 :tags: [skip-execution]
 
-!pip install -q flax
+!pip install -q flax>=0.7.5
 ```
 
 +++ {"id": "b529fbef"}