diff --git a/code_regression.py b/code_regression.py
index aabcd09..2825cf6 100644
--- a/code_regression.py
+++ b/code_regression.py
@@ -90,8 +90,8 @@ def main():
     vocabulary = define_vocabulary('train/in.tsv')
     readed_words = read_input('dev-0/in.tsv')
     readed_words_test_a = read_input('test-A/in.tsv/in.tsv')
-    training(vocabulary, readed_words, 'test.tsv')
-    training(vocabulary, readed_words_test_a, 'test_a.tsv')
+    training(vocabulary, readed_words, 'dev-0/out.tsv')
+    training(vocabulary, readed_words_test_a, 'test-A/out.tsv')
 
 
 # def cost_function(y_hat,y):