diff --git a/day-06/day-06.py b/day-06/day-06.py
index cbd01b1..343afa7 100644
--- a/day-06/day-06.py
+++ b/day-06/day-06.py
@@ -9,7 +9,7 @@ def part_1(input):
         tmp = count[0]
         count = [count[i+1] for i in range(8)]
         count[6] += tmp
-        count[8] = tmp
+        count.append(tmp)
     result = sum(count)
     print("Part 1 result:", result)
 
@@ -21,7 +21,7 @@ def part_2(input):
         tmp = count[0]
         count = [count[i+1] for i in range(8)]
         count[6] += tmp
-        count[8] = tmp
+        count.append(tmp)
     result = sum(count)
     print("Part 2 result:", result)