diff --git a/main.py b/main.py
index 296a66150..d89e24824 100644
--- a/main.py
+++ b/main.py
@@ -701,7 +701,11 @@ def draw_plot_func(dictionary, n_classes, window_title, plot_title, x_label, out
             plt.fill_between(area_under_curve_x, 0, area_under_curve_y, alpha=0.2, edgecolor='r')
             # set window title
             fig = plt.gcf() # gcf - get current figure
-            fig.canvas.set_window_title('AP ' + class_name)
+            try:
+                fig.canvas.set_window_title('AP ' + class_name)
+            except Exception:
+                plt.title('AP ' + class_name)
+
             # set plot title
             plt.title('class: ' + text)
             #plt.suptitle('This is a somewhat long figure title', fontsize=16)