lib/mongoid/elasticsearch/response.rb in mongoid-elasticsearch-0.7.0 vs lib/mongoid/elasticsearch/response.rb in mongoid-elasticsearch-0.8.0

@@ -53,10 +53,15 @@
           case @wrapper
           when :load
             if @multi
               multi_with_load
             else
-              @model.find(hits.map { |h| h['_id'] })
+              records = @model.find(hits.map { |h| h['_id'] })
+              hits.map do |item|
+                records.detect do |record|
+                  record.id.to_s == item['_id'].to_s
+                end
+              end
             end
           when :mash
             hits.map do |h|
               s = h.delete('_source')
               m = Hashie::Mash.new(h.merge(s))