lib/property_sets.rb in property_sets-3.6.0 vs lib/property_sets.rb in property_sets-3.7.0

@@ -8,18 +8,21 @@
 end
 
 module PropertySets
   def self.ensure_property_set_class(association, owner_class_name)
     const_name = "#{owner_class_name}#{association.to_s.singularize.camelcase}".to_sym
+
     unless Object.const_defined?(const_name)
-      property_class = Object.const_set(const_name, Class.new(ActiveRecord::Base))
-      property_class.class_eval do
+      property_class = Class.new(ActiveRecord::Base) do
         include PropertySets::PropertySetModel::InstanceMethods
         extend  PropertySets::PropertySetModel::ClassMethods
       end
 
+      Object.const_set(const_name, property_class)
+
       property_class.owner_class = owner_class_name
       property_class.owner_assoc = association
     end
+
     Object.const_get(const_name)
   end
 end