diff --git a/composer.json b/composer.json
index 4c1b733..78aa658 100644
--- 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
index 3e68ff6..1ea7c41 100644
--- 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));