diff --git a/docs/examples/uncollapsed_vi.py b/docs/examples/uncollapsed_vi.py
index faba8ee1e..3652f63f1 100644
--- a/docs/examples/uncollapsed_vi.py
+++ b/docs/examples/uncollapsed_vi.py
@@ -256,7 +256,7 @@
 # %%
 schedule = ox.warmup_cosine_decay_schedule(
     init_value=0.0,
-    peak_value=0.1,
+    peak_value=0.01,
     warmup_steps=75,
     decay_steps=1500,
     end_value=0.001,