working with classes inherited from ActiveRecord::Base

snop-snov · Nov 18, 2013 · 01a0b42 · 01a0b42
1 parent de1e8d5
commit 01a0b42
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/lib/kaminari/models/active_record_extension.rb b/lib/kaminari/models/active_record_extension.rb
@@ -8,14 +8,14 @@ module ActiveRecordExtension
       class << self
         def inherited_with_kaminari(kls) #:nodoc:
           inherited_without_kaminari kls
-          kls.send(:include, Kaminari::ActiveRecordModelExtension) if kls.superclass == ActiveRecord::Base
+          kls.send(:include, Kaminari::ActiveRecordModelExtension) if kls < ActiveRecord::Base
         end
         alias_method_chain :inherited, :kaminari
       end
 
       # Existing subclasses pick up the model extension as well
       self.descendants.each do |kls|
-        kls.send(:include, Kaminari::ActiveRecordModelExtension) if kls.superclass == ActiveRecord::Base
+        kls.send(:include, Kaminari::ActiveRecordModelExtension) if kls < ActiveRecord::Base
       end
     end
   end