diff --git a/bindings/python/pyiree/jax/README.md b/bindings/python/pyiree/jax/README.md
index d7bf18c..b6b6542 100644
--- a/bindings/python/pyiree/jax/README.md
+++ b/bindings/python/pyiree/jax/README.md
@@ -44,11 +44,10 @@
 
 Install the Android NDK according to the
 [Android Getting Started](https://google.github.io/iree/get-started/getting-started-android-cmake)
-doc, and then ensure the following environment variables are set:
+doc, and then ensure the following environment variable is set:
 
 ```shell
 export ANDROID_NDK=# NDK install location
-export IREE_LLVMAOT_LINKER_PATH="${ANDROID_NDK?}/toolchains/llvm/prebuilt/linux-x86_64/bin/aarch64-linux-android29-clang++"
 ```
 
 The code below assumes that you have `flax` installed.
@@ -62,6 +61,10 @@
 import flax
 from flax import linen as nn
 
+import os
+# Configure the linker to target Android.
+os.environ["IREE_LLVMAOT_LINKER_PATH"] = "${ANDROID_NDK?}/toolchains/llvm/prebuilt/linux-x86_64/bin/aarch64-linux-android29-clang++ -static-libstdc++"
+
 
 class MLP(nn.Module):