protected function _afterLoad()
{
// Resort items according the search response.
$originalItems = $this->_items;
$this->_items = [];
foreach ($this->queryResponse->getIterator() as $document) {
$documentId = $document->getId();
if (isset($originalItems[$documentId])) {
$originalItems[$documentId]->setDocumentScore($document->getScore());
$originalItems[$documentId]->setDocumentSource($document->getSource());
$this->_items[$documentId] = $originalItems[$documentId];
}
}
return parent::_afterLoad();
}