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