diff --git a/src/ReferenceRepository.php b/src/ReferenceRepository.php
index 23adec11..b235a644 100644
--- a/src/ReferenceRepository.php
+++ b/src/ReferenceRepository.php
@@ -196,13 +196,12 @@ public function addReference($name, $object)
     }
 
     /**
-     * Loads an object using stored reference
-     * named by $name
+     * Loads an object using stored reference named by $name
      *
      * @param string $name
      * @psalm-param class-string<T>|null $class
      *
-     * @return object
+     * @return T|object
      * @psalm-return ($class is null ? object : T)
      *
      * @throws OutOfBoundsException - if repository does not exist.