diff --git a/tutorials/logits_lens_with_features.ipynb b/tutorials/logits_lens_with_features.ipynb
index 69edd924..b319cec4 100644
--- a/tutorials/logits_lens_with_features.ipynb
+++ b/tutorials/logits_lens_with_features.ipynb
@@ -130,7 +130,7 @@
     "for layer in range(12):\n",
     "    sae, original_cfg_dict, sparsity = SAE.from_pretrained(\n",
     "        release=\"gpt2-small-res-jb\",\n",
-    "        sae_id=\"blocks.0.hook_resid_pre\",\n",
+    "        sae_id=f\"blocks.{layer}.hook_resid_pre\",\n",
     "        device=\"cpu\",\n",
     "    )\n",
     "    gpt2_small_sparse_autoencoders[f\"blocks.{layer}.hook_resid_pre\"] = sae\n",