diff --git a/research/instahide_attack_2020/step_5_reconstruct.py b/research/instahide_attack_2020/step_5_reconstruct.py
index dcbf6f8..ade5903 100644
--- a/research/instahide_attack_2020/step_5_reconstruct.py
+++ b/research/instahide_attack_2020/step_5_reconstruct.py
@@ -25,7 +25,7 @@ import jax
 import collections
 from PIL import Image
 
-import jax.experimental.optimizers
+import jax.example_libraries.optimizers
 
 import matplotlib.pyplot as plt
 
@@ -109,7 +109,8 @@ def run():
     lams = np.array(lambdas)
 
     # Use Adam, because thinking hard is overrated we have magic pixie dust.
-    init_1, opt_update_1, get_params_1 = jax.experimental.optimizers.adam(.01)
+    init_1, opt_update_1, get_params_1 = \
+        jax.example_libraries.optimizers.adam(.01)
     @jax.jit
     def update_1(i, opt_state, gs):
         return opt_update_1(i, gs, opt_state)