diff --git a/train.py b/train.py
index 9598ce2..6d2ae90 100755
--- a/train.py
+++ b/train.py
@@ -26,7 +26,7 @@ def my_config():
     batch_param = int(sys.argv[1])
     epoch_param = int(sys.argv[2])
 
-def regression_model(epoch_param, batch_param, _run):
+def prepare_model(epoch_param, batch_param, _run):
     _run.info["prepare_model_ts"] = str(datetime.now())
     # odczytanie danych z plików
     vgsales_train = pd.read_csv('train.csv')
@@ -66,7 +66,7 @@ def regression_model(epoch_param, batch_param, _run):
 
 @ex.main
 def my_main(epoch_param, batch_param):
-    print(regression_model())
+    print(prepare_model())
 
 r = ex.run()
 ex.add_artifact("vgsales_model.h5")
\ No newline at end of file