/**
  * {@inheritdoc}
  */
 public function containsKey($key)
 {
     if (!$this->initialized && $this->association['fetch'] === ClassMetadata::FETCH_EXTRA_LAZY && isset($this->association['indexBy'])) {
         $persister = $this->em->getUnitOfWork()->getCollectionPersister($this->association);
         return $this->collection->containsKey($key) || $persister->containsKey($this, $key);
     }
     return parent::containsKey($key);
 }