diff --git a/benchmarks/trace_replay.py b/benchmarks/trace_replay.py
index 5e74f4d3..fbfd093b 100644
--- a/benchmarks/trace_replay.py
+++ b/benchmarks/trace_replay.py
@@ -540,14 +540,16 @@ async def send_group(r=reqs, d=delay):
         flat_requests.extend(reqs)
 
     group_results = await asyncio.gather(*tasks)
+
+    if pbar is not None:
+        pbar.close()
+    await session.close()
+
     outputs = []
     for res in group_results:
         if isinstance(res, list):
             outputs.extend(res)
 
-    if pbar is not None:
-        pbar.close()
-
     benchmark_duration = time.perf_counter() - start_time
     metrics, actual_output_lens = calculate_metrics(
         input_requests=flat_requests,
@@ -678,6 +680,14 @@ def main(args: argparse.Namespace):
 
 
 if __name__ == "__main__":
+    # Check openpyxl for Excel export
+    try:
+        import openpyxl
+    except ImportError:
+        print("\nMissing package: openpyxl")
+        print("Please install openpyxl via pip install.\n")
+        sys.exit(1)
+
     parser = create_argument_trace()
     args = parser.parse_args()
     main(args)