diff --git a/test/benchmark.rb b/test/benchmark.rb
index ae4a167..e8e99c6 100644
--- a/test/benchmark.rb
+++ b/test/benchmark.rb
@@ -12,13 +12,12 @@
 
 def compare_sanitize
   sanitize_config = Sanitize::Config::RELAXED
-  [[DOCUMENT_HUGE, "huge"], [DOCUMENT_MEDIUM, "medium"], [DOCUMENT_SMALL, "small"]].each do |(html, label)|
     Benchmark.ips do |x|
-      x.report("sanitize-document-#{label}") do
+      x.report("sanitize-#{label}") do
         Sanitize.document(html, sanitize_config)
       end
 
-      x.report("selma-document-#{label}") do
+      x.report("selma-#{label}") do
         sanitizer = Selma::Sanitizer.new(Selma::Sanitizer::Config::RELAXED)
         Selma::Rewriter.new(sanitizer: sanitizer).rewrite(html)
       end