diff --git a/test/benchmark/benchmark_comparison.jl b/test/benchmark/benchmark_comparison.jl
index 76db6c0..b958a23 100644
--- a/test/benchmark/benchmark_comparison.jl
+++ b/test/benchmark/benchmark_comparison.jl
@@ -45,7 +45,12 @@ end
 f = Figure(fontsize = 9,);
 axs = [Axis(f[i, j], yscale = log10, xscale = log10) for i in 1:4 for j in 1:2];
 
-labels = ("stream-based", "collection-based", "collection-based without setup")
+labels = (
+    "stream-based\n(StreamSampling.itsample)", 
+    "collection-based with setup\n(StatsBase.sample)", 
+    "collection-based\n(StatsBase.sample)"
+)
+
 markers = (:circle, :rect, :utriangle)
 a, b = 0, 0
 
diff --git a/test/benchmark/comparison.png b/test/benchmark/comparison.png
index 413aeb7..b689360 100644
Binary files a/test/benchmark/comparison.png and b/test/benchmark/comparison.png differ