lib/memento/action_controller_methods.rb in memento-0.5.1 vs lib/memento/action_controller_methods.rb in memento-0.5.2

@@ -13,6 +13,8 @@
     end
     private :memento
   end
 end
 
-ActionController::Base.send(:include, Memento::ActionControllerMethods) if defined?(ActionController::Base)
\ No newline at end of file
+ActiveSupport.on_load(:action_controller) do
+  include Memento::ActionControllerMethods
+end