diff --git a/src/Mapper/TreeMapper.php b/src/Mapper/TreeMapper.php
index b386dde1..8da44254 100644
--- a/src/Mapper/TreeMapper.php
+++ b/src/Mapper/TreeMapper.php
@@ -13,7 +13,8 @@ interface TreeMapper
      * @template T of object
      *
      * @param string|class-string<T> $signature
-     * @return (
+     * @return T
+     * @phpstan-return (
      *     $signature is class-string<T>
      *         ? T
      *         : ($signature is class-string ? object : mixed)