diff --git a/src/Repository/EntitySpecificationRepository.php b/src/Repository/EntitySpecificationRepository.php
index 9bde7d1..7cf1540 100644
--- a/src/Repository/EntitySpecificationRepository.php
+++ b/src/Repository/EntitySpecificationRepository.php
@@ -19,7 +19,7 @@
 /**
  * This class allows you to use a Specification to query entities.
  *
- * @template T
+ * @template T of object
  * @phpstan-extends EntityRepository<T>
  */
 class EntitySpecificationRepository extends EntityRepository implements EntitySpecificationRepositoryInterface
diff --git a/src/Repository/EntitySpecificationRepositoryTrait.php b/src/Repository/EntitySpecificationRepositoryTrait.php
index 3f7d15b..25a7b5d 100644
--- a/src/Repository/EntitySpecificationRepositoryTrait.php
+++ b/src/Repository/EntitySpecificationRepositoryTrait.php
@@ -177,14 +177,7 @@ public function getQueryBuilder($specification, ?string $alias = null): QueryBui
     public function iterate($specification, ?ResultModifier $modifier = null): \Traversable
     {
         $query = $this->getQuery($specification, $modifier);
-
-        if (method_exists($query, 'toIterable')) {
-            yield from $query->toIterable();
-        } else {
-            foreach ($query->iterate() as $key => $row) {
-                yield $key => current($row);
-            }
-        }
+        yield from $query->toIterable();
     }
 
     /**