diff --git a/src/guidellm/benchmark/aggregator.py b/src/guidellm/benchmark/aggregator.py
index 9943f169..af7f1a13 100644
--- a/src/guidellm/benchmark/aggregator.py
+++ b/src/guidellm/benchmark/aggregator.py
@@ -403,7 +403,7 @@ def add_result(
         in_warmup_duration = (
             self.args.warmup_duration
             and result.request_info.worker_start
-            <= (global_start_time - self.args.warmup_duration)
+            <= (global_start_time + self.args.warmup_duration)
         )
 
         if in_warmup_number or in_warmup_duration:
diff --git a/src/guidellm/benchmark/benchmark.py b/src/guidellm/benchmark/benchmark.py
index 4e2e09a3..9f683f8e 100644
--- a/src/guidellm/benchmark/benchmark.py
+++ b/src/guidellm/benchmark/benchmark.py
@@ -816,7 +816,7 @@ def from_stats(
                         for req in total_with_output_first
                     ],
                     iter_counts=[
-                        req.prompt_tokens + req.output_tokens
+                        req.output_tokens
                         for req in total_with_output_first
                     ],
                     first_iter_counts=[