lib/searchkick.rb in searchkick-0.2.3 vs lib/searchkick.rb in searchkick-0.2.4

@@ -7,7 +7,7 @@
 require "searchkick/model"
 require "searchkick/tasks"
 require "searchkick/logger" if defined?(Rails)
 
 # TODO find better ActiveModel hook
-ActiveModel::AttributeMethods::ClassMethods.send(:include, Searchkick::Model)
+ActiveModel::Callbacks.send(:include, Searchkick::Model)
 ActiveRecord::Base.send(:extend, Searchkick::Model) if defined?(ActiveRecord)