diff --git a/torchdrug/data/dataset.py b/torchdrug/data/dataset.py
index 34e9606..e54bdca 100644
--- a/torchdrug/data/dataset.py
+++ b/torchdrug/data/dataset.py
@@ -426,7 +426,7 @@ def load_edge(self, edge_list, node_feature, node_label, node_vocab=None, inv_no
         self.node_vocab = node_vocab
         self.inv_node_vocab = inv_node_vocab
         self.label_vocab = label_vocab
-        self.inv_node_vocab = inv_label_vocab
+        self.inv_label_vocab = inv_label_vocab
 
     def _standarize_vocab(self, vocab, inverse_vocab):
         if vocab is not None: