use to_prepare for Rails 2

attr-encrypted · Jul 22, 2022 · 5a83bc8 · 5a83bc8
1 parent cf8f80b
commit 5a83bc8
Showing 1 changed file with 5 additions and 3 deletions.
diff --git a/lib/attr_encrypted/adapters/active_record.rb b/lib/attr_encrypted/adapters/active_record.rb
@@ -133,8 +133,10 @@ def method_missing_with_attr_encrypted(method, *args, &block)
     end
   end
 
-  ActiveSupport.on_load(:active_record) do
-    extend AttrEncrypted
-    extend AttrEncrypted::Adapters::ActiveRecord
+  Rails.configuration.to_prepare do
+    ActiveRecord::Base.class_eval do
+      extend AttrEncrypted
+      extend AttrEncrypted::Adapters::ActiveRecord
+    end
   end
 end