diff --git a/src/Controller/OgRepository/Webhook/ResponseController.php b/src/Controller/OgRepository/Webhook/ResponseController.php
index beeaf92..cdd7e34 100644
--- a/src/Controller/OgRepository/Webhook/ResponseController.php
+++ b/src/Controller/OgRepository/Webhook/ResponseController.php
@@ -88,7 +88,11 @@ class ResponseController extends AbstractOgRepositoryController
         $imageUuid = $trace->getInput()['imageUuid'];
         $repositoryUuid = $trace->getInput()['repositoryUuid'];
 
-        $image = $this->entityManager->getRepository(Image::class)->findOneBy(['uuid' => $imageUuid]);
+        $imageImageRepository = $this->entityManager->getRepository(ImageImageRepository::class)->findOneBy(['uuid' => $imageUuid]);
+
+        /** @var Image $image */
+        $image = $imageImageRepository->getImage();
+
         $repository = $this->entityManager->getRepository(ImageRepository::class)->findOneBy(['uuid' => $repositoryUuid]);
 
         if ($image === null) {
@@ -98,9 +102,9 @@ class ResponseController extends AbstractOgRepositoryController
 
         $this->logger->info("Image $actionType successful", ['image' => $image->getName()]);
 
-        $image->addRepository($repository);
-        $image->setStatus(ImageStatus::SUCCESS);
-        $this->entityManager->persist($image);
+        $imageImageRepository->setRepository($repository);
+        $imageImageRepository->setStatus(ImageStatus::SUCCESS);
+        $this->entityManager->persist($imageImageRepository);
 
         $this->updateTraceStatus($trace, TraceStatus::SUCCESS);
     }