bundle preparements

dsentker · Sep 12, 2017 · 7954521 · 7954521
1 parent 0b18d1a
commit 7954521
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 5 deletions.
diff --git a/composer.json b/composer.json
@@ -1,7 +1,7 @@
 {
   "name": "dsentker/watcher",
   "description": "Allows to track changes on doctrine entities with an easy-to-use API.",
-  "version": "0.2.42",
+  "version": "0.2.43",
   "license": "MIT",
   "autoload": {
     "psr-4": {

diff --git a/src/Watcher/EventListener/LoadListener.php b/src/Watcher/EventListener/LoadListener.php
@@ -3,7 +3,6 @@
 namespace Watcher\EventListener;
 
 use Doctrine\ORM\Event\LifecycleEventArgs;
-use Watcher\Entity\EntityLog;
 use Watcher\Entity\LogAccessor;
 use Watcher\Repository\EntityLogRepository;
 
@@ -15,12 +14,10 @@ class LoadListener
      */
     public function postLoad(LifecycleEventArgs $args)
     {
-
-        $em = $args->getEntityManager();
         $entity = $args->getEntity();
 
         /** @var EntityLogRepository $logRepo */
-        $logRepo = $em->getRepository(EntityLog::class);
+        $logRepo = $args->getEntityManager()->getRepository(get_class($entity));
 
         if ($entity instanceof LogAccessor) {
             $entity->setLogs($logRepo->getLogsFromEntity($entity));