diff --git a/GenTunic/gen_tuning.py b/GenTunic/gen_tuning.py
index 5a2b160..afb460e 100644
--- a/GenTunic/gen_tuning.py
+++ b/GenTunic/gen_tuning.py
@@ -1,6 +1,8 @@
 
 import math
 import time
+import matplotlib
+matplotlib.use('Agg')
 from matplotlib import pyplot as plt
 import numpy as np
 from GenTunic.gen_math import project_bit
diff --git a/ReinforcmentLearning/learning.py b/ReinforcmentLearning/learning.py
index bd1a85b..1ea08f4 100644
--- a/ReinforcmentLearning/learning.py
+++ b/ReinforcmentLearning/learning.py
@@ -1,3 +1,5 @@
+import matplotlib
+matplotlib.use('Agg')
 import matplotlib.pyplot as plt
 import numpy as np
 import pandas as pd
diff --git a/main.py b/main.py
index 18353ec..f2ee65b 100644
--- a/main.py
+++ b/main.py
@@ -1,7 +1,3 @@
-#! 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