diff --git a/src/ReferenceRepository.php b/src/ReferenceRepository.php
index 0165638b..c75dde17 100644
--- a/src/ReferenceRepository.php
+++ b/src/ReferenceRepository.php
@@ -73,7 +73,7 @@ public function __construct(ObjectManager $manager)
      * @param object $reference Reference object
      * @param object $uow       Unit of work
      *
-     * @return array
+     * @return mixed
      */
     protected function getIdentifier(object $reference, object $uow)
     {
@@ -324,7 +324,7 @@ public function getIdentities()
     /**
      * Get all stored identities
      *
-     * @phpstan-return array<class-string, array<string, object>>
+     * @phpstan-return array<class-string, array<string, mixed>>
      */
     public function getIdentitiesByClass(): array
     {