lib/action_policy/railtie.rb in action_policy-0.1.3 vs lib/action_policy/railtie.rb in action_policy-0.1.4

@@ -47,10 +47,10 @@
     initializer "action_policy.clear_per_thread_cache" do |app|
       app.executor.to_run { ActionPolicy::PerThreadCache.clear_all }
       app.executor.to_complete { ActionPolicy::PerThreadCache.clear_all }
     end
 
-    config.after_initialize do |_app|
+    config.to_prepare do |_app|
       ActiveSupport.on_load(:action_controller) do
         next unless Rails.application.config.action_policy.auto_inject_into_controller
 
         ActionController::Base.include ActionPolicy::Controller