diff --git a/modules/util/plot.py b/modules/util/plot.py
index 2e511639d65436c64e7606699749a9b7163f3f0d..f3313fab9073b69365d0227444cbc5749ef8fc67 100644
--- a/modules/util/plot.py
+++ b/modules/util/plot.py
@@ -25,7 +25,10 @@ img_name = 'train_curve.png'
 
 run_valid_stepping = 1
 
-matplotlib.use('tkagg')
+try:
+    matplotlib.use('tkagg')
+except Exception as e:
+    print(e)
 
 cld_phase_clr_list = [[128,128,128], [11,36,250], [45,254,254], [127,15,126], [254,192,203], [0,0,0]]