lib/graphiti/sideload/has_one.rb in graphiti-1.0.alpha.12 vs lib/graphiti/sideload/has_one.rb in graphiti-1.0.alpha.14

@@ -1,11 +1,11 @@
 class Graphiti::Sideload::HasOne < Graphiti::Sideload::HasMany
   def type
     :has_one
   end
 
-  def assign_each(parent, children)
-    children_hash = children.group_by(&foreign_key)
-    result = children_hash[parent.send(primary_key)] || []
-    result[0]
+  private
+
+  def children_for(parent, map)
+    super[0]
   end
 end