fixing for rails 5 application record compatibility.

ovenbits · Feb 19, 2016 · c784534 · c784534
1 parent a04dfa0
commit c784534
Showing 1 changed file with 4 additions and 3 deletions.
diff --git a/lib/acts_as_follower/follower_lib.rb b/lib/acts_as_follower/follower_lib.rb
@@ -5,10 +5,11 @@ module FollowerLib
 
     # Retrieves the parent class name if using STI.
     def parent_class_name(obj)
-      if obj.class.superclass != ActiveRecord::Base
-        return obj.class.superclass.name
+      if obj.class.superclass == ActiveRecord::Base ||
+        obj.class.superclass.superclass == ActiveRecord::Base
+        return obj.class.name
       end
-      return obj.class.name
+      return obj.class.superclass.name
     end
 
     def apply_options_to_scope(scope, options = {})