lib/mongo_mapper/plugins/modifiers.rb in mongo_mapper-0.8.6 vs lib/mongo_mapper/plugins/modifiers.rb in mongo_mapper-0.9.0

@@ -1,9 +1,11 @@
 # encoding: UTF-8
 module MongoMapper
   module Plugins
     module Modifiers
+      extend ActiveSupport::Concern
+
       module ClassMethods
         def increment(*args)
           modifier_update('$inc', args)
         end
 
@@ -15,10 +17,10 @@
         end
 
         def set(*args)
           criteria, updates = criteria_and_keys_from_args(args)
           updates.each do |key, value|
-            updates[key] = keys[key].set(value) if key?(key)
+            updates[key] = keys[key.to_s].set(value) if key?(key)
           end
           collection.update(criteria, {'$set' => updates}, :multi => true)
         end
 
         def unset(*args)
\ No newline at end of file