diff --git a/GUI_main.py b/GUI_main.py
index f05dbe07cef8449535a0f7f3f5f4cd513caef2e1..a3ee0b37d33ebaaf7b17a92322a747ff2a75b6c6 100644
--- a/GUI_main.py
+++ b/GUI_main.py
@@ -818,10 +818,13 @@ class App(QMainWindow):
                             temp_mask = self.reader.LoadSeg(t, dlg.listfov.row(item))
                             self.reader.SaveMask(t,dlg.listfov.row(item), temp_mask)
             
-            self.m.UpdatePlots()
-            self.ClearStatusBar()
-            self.EnableCNNButtons()
-            self.Enable(self.button_cnn)
+            self.ReloadThreeMasks()
+            
+            
+        self.m.UpdatePlots()
+        self.ClearStatusBar()
+        self.EnableCNNButtons()
+        self.Enable(self.button_cnn)
     
     
     def PredThreshSeg(self, timeindex, fovindex, thr_val, seg_val):