diff --git a/README.md b/README.md
index 2c0261b..20ba3dc 100644
--- a/README.md
+++ b/README.md
@@ -1,6 +1,7 @@
 # KG-BERT: BERT for Knowledge Graph Completion
 
-The repository is modified from [pytorch-pretrained-BERT](https://github.com/huggingface/pytorch-pretrained-BERT).
+The repository is modified from [pytorch-pretrained-BERT](https://github.com/huggingface/pytorch-pretrained-BERT) and tested on Python 3.5+.
+
 
 ## Installing requirement packages
 
@@ -80,7 +81,7 @@ python3 run_bert_relation_prediction.py
 --eval_batch_size 512
 ```
 
-### 2. Link Prediction
+### 3. Link Prediction
 
 #### WN18RR
 
diff --git a/run_bert_triple_classifier.py b/run_bert_triple_classifier.py
index 8bdf9ac..b23b84f 100644
--- a/run_bert_triple_classifier.py
+++ b/run_bert_triple_classifier.py
@@ -117,7 +117,7 @@ class KGProcessor(DataProcessor):
     def get_train_examples(self, data_dir):
         """See base class."""
         return self._create_examples(
-            self._read_tsv(os.path.join(data_dir, "train_10.tsv")), "train", data_dir)
+            self._read_tsv(os.path.join(data_dir, "train.tsv")), "train", data_dir)
 
     def get_dev_examples(self, data_dir):
         """See base class."""