diff --git a/run.py b/run.py
index 55e1dcc..d9f0ff4 100644
--- a/run.py
+++ b/run.py
@@ -75,6 +75,6 @@ class Model():
 
 alpha = float(sys.argv[1])
 print(f'alfa: {alpha}')
-model = Model(alpha, 'dev-0', 'dev-0')
+model = Model(alpha, 'train', 'dev-0')
 model.train()
 model.predict()