lib/rails_erd/domain/specialization.rb in rails-erd-1.4.7 vs lib/rails_erd/domain/specialization.rb in rails-erd-1.5.0

@@ -28,10 +28,10 @@
             new(domain, domain.entity_by_name(model.base_class.name), domain.entity_by_name(model.name))
           }
         end
 
         def abstract_from_models(domain, models)
-          models.select(&:abstract_class?).collect(&:descendants).flatten.collect { |model|
+          models.select(&:abstract_class?).collect(&:direct_descendants).flatten.collect { |model|
             new(domain, domain.entity_by_name(model.superclass.name), domain.entity_by_name(model.name))
           }
         end
       end