diff --git a/tdc/resource/primekg.py b/tdc/resource/primekg.py
index ae0df10a..38935b24 100644
--- a/tdc/resource/primekg.py
+++ b/tdc/resource/primekg.py
@@ -33,9 +33,9 @@ def to_nx(self):
 
         G = nx.Graph()
         for i in self.df.relation.unique():
-            G.add_edges_from(self.df[self.df.relation == i][["x_id",
-                                                             "y_id"]].values,
-                             relation=i)
+            G.add_edges_from(
+                self.df[self.df.relation == i][["x_name", "y_name"]].values,
+                relation=i)
         return G
 
     def get_features(self, feature_type):