lib/with_model/model.rb in with_model-2.1.6 vs lib/with_model/model.rb in with_model-2.1.7

@@ -54,10 +54,10 @@
     end
 
     def cleanup_descendants_tracking
       if defined?(ActiveSupport::DescendantsTracker)
         if ActiveSupport::VERSION::MAJOR >= 7
-          ActiveSupport::DescendantsTracker.clear([ActiveRecord::Base])
+          ActiveSupport::DescendantsTracker.clear([@model])
         else
           ActiveSupport::DescendantsTracker.class_variable_get(:@@direct_descendants).delete(ActiveRecord::Base)
         end
       elsif @model.superclass.respond_to?(:direct_descendants)
         @model.superclass.direct_descendants.delete(@model)