diff --git a/lib/bugsnag/railtie.rb b/lib/bugsnag/railtie.rb
index 989c9e21a..799373b31 100644
--- a/lib/bugsnag/railtie.rb
+++ b/lib/bugsnag/railtie.rb
@@ -37,12 +37,9 @@ class Railtie < Rails::Railtie
       config = YAML.load_file(config_file) if File.exist?(config_file)
       Bugsnag.configure(config[::Rails.env] ? config[::Rails.env] : config) if config
 
-      if defined?(::ActionController::Base)
+      ActiveSupport.on_load(:action_controller) do
         require "bugsnag/rails/controller_methods"
-        ::ActionController::Base.send(:include, Bugsnag::Rails::ControllerMethods)
-      end
-      if defined?(ActionController::API)
-        ActionController::API.send(:include, Bugsnag::Rails::ControllerMethods)
+        include Bugsnag::Rails::ControllerMethods
       end
       ActiveSupport.on_load(:active_record) do
         require "bugsnag/rails/active_record_rescue"