1 files changed, 2 insertions, 2 deletions
diff --git a/src/tests/aoc_2021_09/benches/main.rs b/src/tests/aoc_2021_09/benches/main.rs
index cc9b460..90d22f1 100644
--- a/src/tests/aoc_2021_09/benches/main.rs
+++ b/src/tests/aoc_2021_09/benches/main.rs
@@ -12,7 +12,7 @@ fn run_benches(c: &mut Criterion) {
                 b.iter_batched_ref(
                     || HeightMap::gen(&mut rng, n),
                     |map| map.part1(),
-                    BatchSize::SmallInput,
+                    BatchSize::NumIterations(1),
                 )
             },
         );
@@ -24,7 +24,7 @@ fn run_benches(c: &mut Criterion) {
                 b.iter_batched_ref(
                     || HeightMap::gen(&mut rng, n),
                     |map| map.part2(),
-                    BatchSize::SmallInput,
+                    BatchSize::NumIterations(1),
                 )
             },
         );