diff --git a/main.py b/main.py
index cdb9b81..18353ec 100644
--- a/main.py
+++ b/main.py
@@ -1,3 +1,7 @@
+#! TMP
+import os
+os.environ['MPLBACKEND'] = 'Agg'
+
 from GenTunic.gen_tuning import gen_tuning_main
 from ReinforcmentLearning.learning import multipleTries, oneTry
 from ReinforcmentLearning.util import calc_time_reward
@@ -18,9 +22,7 @@ plot_result = False
 show_game = False
 
 
-#! TMP
-import os
-os.environ['MPLBACKEND'] = 'Agg'
+
 
 
 # oneTry(EPSILON, ALPHA, GAMMA, AMOUNT_RUNS, REWARD_ON_WIN, REWARD_ON_LOSE, plot_result, show_game)