From 80538175e887cfd88dbcb7bbc32bad7c2945b842 Mon Sep 17 00:00:00 2001
From: Pascal Seitz <pascal.seitz@gmail.com>
Date: Wed, 16 Oct 2024 10:33:24 +0800
Subject: [PATCH] fix build

---
 benches/agg_bench.rs | 4 +++-
 1 file changed, 3 insertions(+), 1 deletion(-)

diff --git a/benches/agg_bench.rs b/benches/agg_bench.rs
index 1a2322cd1..2690ea7ab 100644
--- a/benches/agg_bench.rs
+++ b/benches/agg_bench.rs
@@ -1,3 +1,4 @@
+use binggan::plugins::PeakMemAllocPlugin;
 use binggan::{black_box, InputGroup, PeakMemAlloc, INSTRUMENTED_SYSTEM};
 use rand::prelude::SliceRandom;
 use rand::rngs::StdRng;
@@ -45,7 +46,8 @@ fn main() {
 }
 
 fn bench_agg(mut group: InputGroup<Index>) {
-    group.set_alloc(GLOBAL); // Set the peak mem allocator. This will enable peak memory reporting.
+    group.add_plugin(PeakMemAllocPlugin::new(GLOBAL));
+
     register!(group, average_u64);
     register!(group, average_f64);
     register!(group, average_f64_u64);