public function getRepository(EntityManagerInterface $entityManager, $entityName) { $repository = parent::getRepository($entityManager, $entityName); if ($repository instanceof NestedTreeRepository) { $repository->setTreeListener($this->serviceLocator->getByType(TreeListener::class)); } return $repository; }