diff --git a/src/Resources/views/Generator/AssociationTemplate.html.twig b/src/Resources/views/Generator/AssociationTemplate.html.twig
index 0e224e9..e11d0f2 100644
--- a/src/Resources/views/Generator/AssociationTemplate.html.twig
+++ b/src/Resources/views/Generator/AssociationTemplate.html.twig
@@ -135,6 +135,10 @@
         return $qb;
     }
 
+    /**
+     * @return \{{ entityClasspath }}[]
+     *
+     **/
     public function findBy{{ column }}(
         mixed $value,
         bool $useQueryCache = false,
diff --git a/tests/Repository/MyClassRepositoryBase.php b/tests/Repository/MyClassRepositoryBase.php
index 3213271..950e22c 100644
--- a/tests/Repository/MyClassRepositoryBase.php
+++ b/tests/Repository/MyClassRepositoryBase.php
@@ -925,6 +925,10 @@ public static function filterNotInForeignClasses(
         return $qb;
     }
 
+    /**
+     * @return \Tbn\QueryBuilderRepositoryGeneratorBundle\Tests\Entity\MyClass[]
+     *
+     **/
     public function findByForeignClasses(
         mixed $value,
         bool $useQueryCache = false,
diff --git a/tests/Service/ExpectedMyClassRepositoryBase.txt b/tests/Service/ExpectedMyClassRepositoryBase.txt
index 3213271..950e22c 100644
--- a/tests/Service/ExpectedMyClassRepositoryBase.txt
+++ b/tests/Service/ExpectedMyClassRepositoryBase.txt
@@ -925,6 +925,10 @@ class MyClassRepositoryBase extends \Doctrine\Bundle\DoctrineBundle\Repository\S
         return $qb;
     }
 
+    /**
+     * @return \Tbn\QueryBuilderRepositoryGeneratorBundle\Tests\Entity\MyClass[]
+     *
+     **/
     public function findByForeignClasses(
         mixed $value,
         bool $useQueryCache = false,