diff --git a/lib/spree_sitemap_generator.rb b/lib/spree_sitemap_generator.rb
index 63da9ee..ce4cb71 100644
--- a/lib/spree_sitemap_generator.rb
+++ b/lib/spree_sitemap_generator.rb
@@ -25,7 +25,7 @@ module SpreeSitemapGenerator
       SitemapGenerator::Sitemap.default_host = "http://#{Spree::Config[:site_url]}"
 
       require 'spree_sitemap_generator/spree_defaults'
-      SitemapGenerator::LinkSet.send :include, SpreeSitemapGenerator::SpreeDefaults
+      SitemapGenerator::Interpreter.send :include, SpreeSitemapGenerator::SpreeDefaults
 
       Dir.glob(File.join(File.dirname(__FILE__), "../app/**/*_decorator*.rb")) do |c|
         Rails.env.production? ? require(c) : load(c)
diff --git a/spree_sitemap_generator.gemspec b/spree_sitemap_generator.gemspec
index b1cdb08..f01513e 100644
--- a/spree_sitemap_generator.gemspec
+++ b/spree_sitemap_generator.gemspec
@@ -15,8 +15,7 @@ Gem::Specification.new do |s|
   s.require_path = 'lib'
   s.requirements << 'none'
 
-  s.has_rdoc = true
 
   s.add_dependency('spree_core', '>= 0.30.1')
-  s.add_dependency('sitemap_generator', '>= 1.3.4')
+  s.add_dependency('sitemap_generator', '= 2.0.1')
 end