diff --git a/GUI_main.py b/GUI_main.py
index d394eb8c26610a1c2f8ad4654fa0a8b486f8c97a..4ad4499cd36a35fe214449256d3db89aa60cecaa 100644
--- a/GUI_main.py
+++ b/GUI_main.py
@@ -814,10 +814,7 @@ class App(QMainWindow):
                             temp_mask = self.reader.LoadSeg(t, dlg.listfov.row(item))
                             self.reader.SaveMask(t,dlg.listfov.row(item), temp_mask)
             
-            self.ReloadThreeMasks()
-               
-            #once it has iterated over all the fov, the message in 
-            #the status bar is cleared and the buttons are enabled.
+            self.m.UpdatePlots()
             self.ClearStatusBar()
             self.EnableCNNButtons()
    
@@ -1691,7 +1688,7 @@ class App(QMainWindow):
         self.button_exval.setEnabled(False)
         self.button_changecellvalue.setEnabled(False)
         self.button_showval.setEnabled(False)
-    
+
     
     def ButtonSaveMask(self):
         """