lib/effective_roles/engine.rb in effective_roles-2.3.0 vs lib/effective_roles/engine.rb in effective_roles-2.3.1

@@ -2,10 +2,10 @@
   class Engine < ::Rails::Engine
     engine_name 'effective_roles'
 
     # Include acts_as_addressable concern and allow any ActiveRecord object to call it
     initializer 'effective_roles.active_record' do |app|
-      ActiveSupport.on_load :active_record do
+      app.config.to_prepare do
         ActiveRecord::Base.extend(ActsAsRoleRestricted::Base)
       end
     end
 
     # Set up our default configuration options.