diff --git a/invesalius/gui/data_notebook.py b/invesalius/gui/data_notebook.py
index 3e96a24..ed5dcc3 100644
--- a/invesalius/gui/data_notebook.py
+++ b/invesalius/gui/data_notebook.py
@@ -445,10 +445,12 @@ class MasksListCtrlPanel(wx.ListCtrl, listmix.TextEditMixin):
                 self.SetItemImage(key, 0)
 
     def __hide_current_mask(self, pubsub_evt):
-        self.SetItemImage(self.current_index, 0)
+        if self.mask_list_index:
+            self.SetItemImage(self.current_index, 0)
 
     def __show_current_mask(self, pubsub_evt):
-        self.SetItemImage(self.current_index, 1)
+        if self.mask_list_index:
+            self.SetItemImage(self.current_index, 1)
 
     def __init_columns(self):
 
--
libgit2 0.21.2