lib/rambulance/railtie.rb in rambulance-0.6.0 vs lib/rambulance/railtie.rb in rambulance-1.0.0

@@ -1,16 +1,19 @@
 module Rambulance
   class Railtie < Rails::Railtie
     initializer 'rambulance', after: :prepend_helpers_path do |app|
-      require "rambulance/exceptions_app"
+      ActiveSupport.on_load(:action_controller) do
+        require "rambulance/exceptions_app"
+      end
 
-      app.config.exceptions_app =
+      app.config.exceptions_app = ->(env) {
         begin
           ActiveSupport::Dependencies.load_missing_constant(Object, :ExceptionsApp)
-          ->(env){ ::ExceptionsApp.call(env) }
+          ::ExceptionsApp.call(env)
         rescue NameError
-          ::Rambulance::ExceptionsApp
+          ::Rambulance::ExceptionsApp.call(env)
         end
+      }
 
       ActiveSupport.on_load(:after_initialize) do
         Rails.application.routes.append do
           mount app.config.exceptions_app, at: '/rambulance'
         end if Rails.env.development?