diff --git a/causallearn/graph/GraphClass.py b/causallearn/graph/GraphClass.py
index af5bf32b..4d6567a6 100644
--- a/causallearn/graph/GraphClass.py
+++ b/causallearn/graph/GraphClass.py
@@ -153,6 +153,7 @@ def to_nx_graph(self):
         nodes = range(len(self.G.graph))
         self.labels = {i: self.G.nodes[i].get_name() for i in nodes}
         self.nx_graph.add_nodes_from(nodes)
+        self.nx_graph = nx.relabel_nodes(self.nx_graph, self.labels)
         undirected = self.find_undirected()
         directed = self.find_fully_directed()
         bidirected = self.find_bi_directed()