diff --git a/lectures/back_prop.md b/lectures/back_prop.md
index 0f9836811..c8b5eac16 100644
--- a/lectures/back_prop.md
+++ b/lectures/back_prop.md
@@ -16,7 +16,7 @@ kernelspec:
 ```{code-cell} ipython3
 :tags: [hide-output]
 
-!pip install --upgrade jax jaxlib
+!conda install -y -c conda-forge jax jaxlib
 !conda install -y -c plotly plotly plotly-orca retrying
 ```
 
@@ -599,4 +599,5 @@ print(xla_bridge.get_backend().platform)
 This lecture site is built in a server environment that doesn't have access to a `gpu`
 If you run this lecture locally this lets you know where your code is being executed, either
 via the `cpu` or the `gpu`
-```
\ No newline at end of file
+```
+