diff --git a/Chapter06/01_Social_network_analysis.ipynb b/Chapter06/01_Social_network_analysis.ipynb
index 535a774..2455a71 100644
--- a/Chapter06/01_Social_network_analysis.ipynb
+++ b/Chapter06/01_Social_network_analysis.ipynb
@@ -1207,7 +1207,7 @@
     "from node2vec.edges import HadamardEmbedder \n",
     "from stellargraph.data import EdgeSplitter \n",
     "\n",
-    "node2vec = Node2Vec(graph_train) \n",
+    "node2vec = Node2Vec(G) \n",
     "model = node2vec.fit() \n",
     "edges_embs = HadamardEmbedder(keyed_vectors=model.wv) \n",
     "train_embeddings = [edges_embs[str(x[0]),str(x[1])] for x in samples_train]\n",