diff --git a/tests/perf_test/stream/stream_speed/benchmark.py b/tests/perf_test/stream/stream_speed/benchmark.py
index fd5aebb0..c817ebf9 100644
--- a/tests/perf_test/stream/stream_speed/benchmark.py
+++ b/tests/perf_test/stream/stream_speed/benchmark.py
@@ -1,3 +1,5 @@
+"""Consume 10K tokens from the stream endpoint and measure the speed."""
+
 import logging
 import time