diff --git a/benchmark.py b/benchmark.py
index 4148687..9e056b8 100644
--- a/benchmark.py
+++ b/benchmark.py
@@ -28,7 +28,7 @@ args = {
 
 
 def run_benchmark(batch: int, extra: bool):
-    shared.state.begin()
+    shared.state.begin('benchmark')
     if args['sd_model'] is None:
         args['sd_model'] = shared.sd_model
     args['batch_size'] = batch