diff --git a/docs/notebooks/basics.ipynb b/docs/notebooks/basics.ipynb
index 0995edf85..b421b1334 100644
--- a/docs/notebooks/basics.ipynb
+++ b/docs/notebooks/basics.ipynb
@@ -234,7 +234,7 @@
         "forward_without_rng = hk.without_apply_rng(hk.transform(_forward_fn_linear1))\n",
         "params = forward_without_rng.init(rng=rng_key, x=sample_x)\n",
         "output = forward_without_rng.apply(x=sample_x, params=params)\n",
-        "print(f'Output without random key in forward pass \\n {output_1}')"
+        "print(f'Output without random key in forward pass \\n {output}')"
       ]
     },
     {