lib/alba/nested_attribute.rb in alba-2.0.0 vs lib/alba/nested_attribute.rb in alba-2.0.1

@@ -8,11 +8,12 @@
       @block = block
     end
 
     # @return [Hash]
     def value(object)
-      resource_class = Alba.resource_class(&@block)
+      resource_class = Alba.resource_class
       resource_class.transform_keys(@key_transformation)
+      resource_class.class_eval(&@block)
       resource_class.new(object).serializable_hash
     end
   end
 end