diff --git a/train.py b/train.py
index 56ca6d8..21dd0cc 100644
--- a/train.py
+++ b/train.py
@@ -14,13 +14,13 @@ exint.observers.append(FileStorageObserver('my_runs'))
 @exint.config
 def my_config():
     EPOCHS = int(os.environ['EPOCHS'])
-    train_data_x = exint.open_resource('./X_train.csv', 'r')
     
 
 @exint.main
-def my_main(EPOCHS,train_data_x):
+def my_main(EPOCHS):
     _run.info["epochs"] = EPOCHS
     normalize = layers.Normalization()
+    train_data_x = exint.open_resource('./X_train.csv', 'r')
     games_all = train_data_x.copy()
     games_predict = train_data_x.pop('User_Score')
     normalize.adapt(games_all)