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