/**
  * @param string $key
  *
  * @return \Iterator
  */
 protected function getEntityData($key)
 {
     $this->ensureEntityRegistered($key);
     return $this->templateManager->getEntityRegistry()->getData($this->templateManager, $this->getEntityClass(), $key);
 }