diff --git a/src/AbstractLazyCollection.php b/src/AbstractLazyCollection.php
index e7a2d686..2463cfc3 100644
--- a/src/AbstractLazyCollection.php
+++ b/src/AbstractLazyCollection.php
@@ -177,6 +177,11 @@ public function findFirst(Closure $p): mixed
         return $this->collection->findFirst($p);
     }
 
+    /**
+     * {@inheritDoc}
+     *
+     * @psalm-return Collection<TKey,T>
+     */
     public function filter(Closure $p): Collection
     {
         $this->initialize();
@@ -191,6 +196,15 @@ public function forAll(Closure $p): bool
         return $this->collection->forAll($p);
     }
 
+    /**
+     * {@inheritDoc}
+     *
+     * @psalm-param Closure(T):U $func
+     *
+     * @psalm-return Collection<TKey,U>
+     *
+     * @psalm-template U
+     */
     public function map(Closure $func): Collection
     {
         $this->initialize();