lib/secure_headers/railtie.rb in secure_headers-1.3.3 vs lib/secure_headers/railtie.rb in secure_headers-1.3.4

@@ -1,11 +1,15 @@
 # rails 3.1+
 if defined?(Rails::Railtie)
   module SecureHeaders
     class Railtie < Rails::Engine
       isolate_namespace ::SecureHeaders if defined? isolate_namespace # rails 3.0
-      ActionController::Base.send :include, ::SecureHeaders
+      initializer "secure_headers.action_controller" do
+        ActiveSupport.on_load(:action_controller) do
+          include ::SecureHeaders
+        end
+      end
     end
   end
 else
   module ActionController
     class Base
@@ -32,6 +36,6 @@
   end
 
   if defined? ActionController::Routing
     ActionController::Routing::RouteSet::Mapper.send :include, ::SecureHeaders::Routing::MapperExtensions
   end
-end
\ No newline at end of file
+end