./lib/pjax-rails.rb in pjax-rails-0.1.3 vs ./lib/pjax-rails.rb in pjax-rails-0.1.4

@@ -1,11 +1,9 @@
 require 'pjax'
 
 module PjaxRails
   class Engine < ::Rails::Engine
     initializer "pjax_rails.add_controller" do
-      ActiveSupport.on_load(:action_controller) do
-        ::ApplicationController.send :include, Pjax
-      end
+      config.to_prepare { ApplicationController.send :include, Pjax }
     end
   end
 end
\ No newline at end of file