factory.rb in custom_fields-2.4.0.rc1

- old
+ new

@@ -2,19 +2,18 @@
 module Mongoid #:nodoc:
 
   # Instantiates documents that came from the database.
   module Factory
 
-    def from_db_with_custom_fields(klass, attributes = {}, criteria_instance_id = nil)
+    def from_db_with_custom_fields(klass, attributes = nil, selected_fields = nil)
       if klass.with_custom_fields?
         klass.klass_with_custom_fields(attributes['custom_fields_recipe'])
       end
-      from_db_without_custom_fields(klass, attributes, criteria_instance_id)
+
+      from_db_without_custom_fields(klass, attributes, selected_fields)
     end
 
     # equivalent for "alias_method_chain :from_db, :custom_fields"
     alias_method :from_db_without_custom_fields, :from_db unless method_defined?(:from_db_without_custom_fields)
     alias_method :from_db, :from_db_with_custom_fields
-
   end
-
 end