lib/neo4j/active_node/labels.rb in neo4j-5.0.14 vs lib/neo4j/active_node/labels.rb in neo4j-5.0.15

@@ -78,9 +78,15 @@
       protected
 
       module ClassMethods
         include Neo4j::ActiveNode::QueryMethods
 
+        def before_remove_const
+          associations.each_value(&:queue_model_refresh!)
+          MODELS_FOR_LABELS_CACHE.clear
+          WRAPPED_CLASSES.clear
+        end
+
         # Returns the object with the specified neo4j id.
         # @param [String,Integer] id of node to find
         def find(id)
           map_id = proc { |object| object.respond_to?(:id) ? object.send(:id) : object }