diff --git a/ipi_extract.py b/ipi_extract.py
index 04ab46f7129510774fe342a9a937041d99dbeb1c..22491af18c8f541ddb9b544183cd33eb2b44ea6a 100644
--- a/ipi_extract.py
+++ b/ipi_extract.py
@@ -311,11 +311,7 @@ class Callback(object):
             self.curr_vert[self.curr] = 0
             self.cursor[self.curr].linev[0].set_linestyle('--')
             self.cursor[self.curr].linev[self.cursor[self.curr].num_cur].set_linestyle('--')
-            for i in range(4):
-                if i < 2:
-                    ax_group[i].set_visible(False)
-                else:
-                    ax_group[i].set_visible(False)
+            ax_group.set_visible(False)
         else:
             self.curr_ind[self.curr] = np.argmax(mpos / FSSR == self.offset[:, 0])
             c = np.array([0, 0, 0, 0.])
@@ -537,6 +533,10 @@ class Callback(object):
         self.df[self.offset[self.curr_ind[self.curr], 0]] = EMLN.copy()
         self._set_label()
         self._set_visible()
+        self.curr_vert[self.curr] = 0
+        self.cursor[self.curr].linev[0].set_linestyle('--')
+        self.cursor[self.curr].linev[self.cursor[self.curr].num_cur].set_linestyle('--')
+        self.view_data[self.curr].set_visible(False)
         plt.draw()
 
     def reset(self, song, sr, song_resample):