diff --git a/tools.py b/tools.py
index 73ba1c09de1d4a74c8e92449a42a36e0bd4f272c..95d05138defc2076b7f23b4040768f3b8e6e3f2f 100644
--- a/tools.py
+++ b/tools.py
@@ -246,17 +246,16 @@ class Memory:
 
         if self.i == settings.NUM_ITERATIONS:
             list_len = []
-            count_copy = self.count.copy()
-            for i, val in enumerate(count_copy):
+            for i, val in enumerate(self.count):
                 if 0 in val:
                     val.remove(0)
                 if len(val) == 0:
                     del self.memory[i]
-                    del self.count[i]
                 else:
                     val = [val[i] - 1 for i in range(len(val))]
                     self.count[i] = val
                     list_len.append(len(val))
+            self.count = [i for i in self.count if len(i) > 0]
             self.i -= 1
             self.max_idx = np.argmax(list_len)