diff --git a/roberta_ft/04_predict.py b/roberta_ft/04_predict.py
index 2e45f57..1eb9956 100644
--- a/roberta_ft/04_predict.py
+++ b/roberta_ft/04_predict.py
@@ -19,7 +19,7 @@ def write(f_path_in, f_path_out):
             a = get_formatted(l_in)
             f_out.write(a + '\n')
 
-model = 'robertamodel/checkpoint-24000'
+model = 'robertamodel'
 unmasker = pipeline('fill-mask', model=model)
 write('dev-0_in.csv', '../dev-0/out.tsv')
 write('test-A_in.csv', '../test-A/out.tsv')
diff --git a/roberta_ft/run.sh b/roberta_ft/run.sh
index ec35538..a35e620 100644
--- a/roberta_ft/run.sh
+++ b/roberta_ft/run.sh
@@ -1,3 +1,3 @@
 python 00_create_datasets.py
-python 04_predict.py
 bash 2_run.sh
+python 04_predict.py
diff --git a/roberta_with_year_ft/04_predict.py b/roberta_with_year_ft/04_predict.py
index 622616f..1eb9956 100644
--- a/roberta_with_year_ft/04_predict.py
+++ b/roberta_with_year_ft/04_predict.py
@@ -19,7 +19,7 @@ def write(f_path_in, f_path_out):
             a = get_formatted(l_in)
             f_out.write(a + '\n')
 
-model = 'robertamodel/checkpoint-60000'
+model = 'robertamodel'
 unmasker = pipeline('fill-mask', model=model)
 write('dev-0_in.csv', '../dev-0/out.tsv')
 write('test-A_in.csv', '../test-A/out.tsv')