From 695d43748caa067ff99908fe5de37ee99a06d930 Mon Sep 17 00:00:00 2001
From: Thiago Franco de Moraes <totonixsame@gmail.com>
Date: Fri, 19 Mar 2021 19:24:46 -0300
Subject: [PATCH] It was cleaning every mask when removing a mask

---
 invesalius/project.py | 14 +++++++-------
 1 file changed, 7 insertions(+), 7 deletions(-)

diff --git a/invesalius/project.py b/invesalius/project.py
index aa30aef..fad1ad6 100644
--- a/invesalius/project.py
+++ b/invesalius/project.py
@@ -119,14 +119,14 @@ class Project(metaclass=Singleton):
 
     def RemoveMask(self, index):
         new_dict = {}
+        new_index = 0
         for i in self.mask_dict:
-            mask = self.mask_dict[i]
-            mask.cleanup()
-            if i < index:
-                new_dict[i] = self.mask_dict[i]
-            if i > index:
-                new_dict[i-1] = self.mask_dict[i]
-                new_dict[i-1].index = i-1
+            if i == index:
+                mask = self.mask_dict[i]
+                mask.cleanup()
+            else:
+                new_dict[new_index] = self.mask_dict[i]
+                new_index += 1
         self.mask_dict = new_dict
 
     def GetMask(self, index):
--
libgit2 0.21.2