diff --git a/generator.py b/generator.py
index 0a7545a..81c6471 100644
--- a/generator.py
+++ b/generator.py
@@ -9,7 +9,7 @@ min_num = 0
 max_num = 50
 nums = range(min_num, max_num+1)
 signs = ['+', '-', '/', '*']
-num_of_ifs = len(signs)*(max_num-min_num)**2
+num_of_ifs = len(signs)*(max_num-min_num+1)**2
 
 print("""# my_first_calculator.py by AceLewis
 # TODO: Make it work for all floating point numbers too