module ActiveSupport::DescendantsTracker

def direct_descendants(klass)
  descendants = @@direct_descendants[klass]
  descendants ? descendants.to_a : []
end