diff --git a/lib/acts_as_follower/follower_lib.rb b/lib/acts_as_follower/follower_lib.rb index 2bd01b4..f681930 100644 --- 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 = {})