lib/slugs/railtie.rb in slugs-1.3.2 vs lib/slugs/railtie.rb in slugs-2.0.0

@@ -1,9 +1,17 @@
 module Slugs
   class Railtie < Rails::Railtie
 
     initializer 'slugs' do
-      ::ActiveRecord::Base.send :include, Slugs::ActiveRecord::Base
+      ::ActionDispatch::Routing::RouteSet::NamedRouteCollection::UrlHelper::OptimizedUrlHelper.prepend(
+        Slugs::Extensions::ActionDispatch::OptimizedUrlHelper
+      )
+      ::ActionDispatch::Routing::RouteSet::Generator.prepend(
+        Slugs::Extensions::ActionDispatch::Generator
+      )
+      ::ActiveRecord::Base.include(
+        Slugs::Extensions::ActiveRecord::Base
+      )
     end
 
   end
 end