lib/mongomapper/associations/belongs_to_proxy.rb in djsun-mongomapper-0.3.1.1 vs lib/mongomapper/associations/belongs_to_proxy.rb in djsun-mongomapper-0.3.3

@@ -5,17 +5,17 @@
         if doc
           doc.save if doc.new?
           id = doc.id
         end
         
-        @owner.send("#{@association.belongs_to_key_name}=", id)
+        @owner.send("#{@association.foreign_key}=", id)
         reset
       end
 
-      protected
+      protected        
         def find_target
-          if association_id = @owner.send(@association.belongs_to_key_name)
-            @association.klass.find(association_id)
+          if association_id = @owner.send(@association.foreign_key)
+            @association.klass.find_by_id(association_id)
           end
         end
     end
   end
 end