diff --git a/docs/source/notebooks/Notebook_2_LP_Pipeline.ipynb b/docs/source/notebooks/Notebook_2_LP_Pipeline.ipynb
index 1e21a4e82e..c026fa21bf 100644
--- a/docs/source/notebooks/Notebook_2_LP_Pipeline.ipynb
+++ b/docs/source/notebooks/Notebook_2_LP_Pipeline.ipynb
@@ -94,8 +94,8 @@
     "    num_negative_edges=10,\n",
     "    node_feats=nfeats_4_modeling,\n",
     "    batch_size=64,\n",
-    "    exclude_training_targets=False,\n",
-    "    reverse_edge_types_map=[\"paper,citing,cited,paper\"],\n",
+    "    exclude_training_targets=True,\n",
+    "    reverse_edge_types_map={(\"paper\", \"citing\", \"paper\"):(\"paper\",\"cited\",\"paper\")},\n",
     "    train_task=True)\n",
     "val_dataloader = gs.dataloading.GSgnnLinkPredictionTestDataLoader(\n",
     "    dataset=acm_data,\n",