diff --git a/src/ProphecyTrait.php b/src/ProphecyTrait.php
index 7c9ada2..db1676f 100644
--- a/src/ProphecyTrait.php
+++ b/src/ProphecyTrait.php
@@ -36,7 +36,10 @@ trait ProphecyTrait
      * @throws DoubleException
      * @throws InterfaceNotFoundException
      *
-     * @psalm-param class-string|null $classOrInterface
+     * @template T of object
+     * @phpstan-param class-string<T>|null $classOrInterface
+     * @phpstan-return ($classOrInterface is null ? ObjectProphecy<object> : ObjectProphecy<T>)
+     *
      * @not-deprecated
      */
     protected function prophesize(?string $classOrInterface = null): ObjectProphecy