public function update(Document $document, array $options = [])
{
$documentId = $this->getDocumentId($document);
return $this->updateById($document, $documentId, $document, $options);
}

It seems the first arg to call to updateById should be $collection, and the update function does not accept $collection. Am I missing something?