diff --git a/Chapter03/05_GraphAutoEncoder_StellarGraph.ipynb b/Chapter03/05_GraphAutoEncoder_StellarGraph.ipynb
index c93b1bf..e080818 100644
--- a/Chapter03/05_GraphAutoEncoder_StellarGraph.ipynb
+++ b/Chapter03/05_GraphAutoEncoder_StellarGraph.ipynb
@@ -60,7 +60,7 @@
     "\n",
     "edge_splitter_train = EdgeSplitter(G_test)\n",
     "\n",
-    "G_train, edge_ids_train, edge_labels_train = edge_splitter_test.train_test_split(\n",
+    "G_train, edge_ids_train, edge_labels_train = edge_splitter_train.train_test_split(\n",
     "    p=0.1, method=\"global\", keep_connected=True\n",
     ")"
    ]