diff --git a/README.md b/README.md
index 3993eda..68df8ff 100644
--- a/README.md
+++ b/README.md
@@ -59,10 +59,10 @@ python /utils/build_graphs.py --mode MODE --cpus NUM_CPUS
 `MODE` can be `train`, `val` or `test`. NOTE: check `utils/build_graphs.py` to make sure you're loading the correct dataset to generate the graphs you want. 
 
 ## Training
-You can use the `gtbc.sh`.
+Use `run_train.sh`.
 
 ## Testing 
-Use `run_test_tom.sh`. 
+Use `run_test.sh`. 
 
 # Hardware setup
 All models are trained on an NVIDIA Tesla V100-SXM2-32GB GPU.