diff --git a/day01/part2.py b/day01/part2.py
index 1cc188b..8f2c0b9 100644
--- a/day01/part2.py
+++ b/day01/part2.py
@@ -5,7 +5,7 @@ items = [int(x.strip()) for x in open("input.txt")]
 
 COUNT = 3
 
-for vals in combinations(items, 3):
+for vals in combinations(items, COUNT):
     if sum(vals) == 2020:
         print("Found [{}]: Sum={}, Product={}".format(
             ", ".join(map(str, vals)),