diff --git a/benchmark/benchmark.rb b/benchmark/benchmark.rb
index <HASH>..<HASH> 100755
--- a/benchmark/benchmark.rb
+++ b/benchmark/benchmark.rb
@@ -18,7 +18,7 @@ def compare_scrub_methods
   puts Loofah.scrub_fragment(snip, :escape).to_s
   puts "--"
   puts HTMLFilter.new.filter(snip)
-  puts Loofah::Helpers.sanitize(snip)
+  puts Loofah.scrub_fragment(snip, :strip).to_s
   puts
 end
 
diff --git a/benchmark/helper.rb b/benchmark/helper.rb
index <HASH>..<HASH> 100644
--- a/benchmark/helper.rb
+++ b/benchmark/helper.rb
@@ -9,6 +9,10 @@ require "sanitize"
 require 'hitimes'
 require 'htmlfilter'
 
+unless defined?(HTMLFilter)
+  HTMLFilter = HtmlFilter
+end
+
 class RailsSanitize
   include ActionView::Helpers::SanitizeHelper
   extend ActionView::Helpers::SanitizeHelper::ClassMethods
fixing trans's commit to work with the current gem class names.