diff --git a/stubs/EntityRepository.phpstub b/stubs/EntityRepository.phpstub
index c132850..e0dc548 100644
--- a/stubs/EntityRepository.phpstub
+++ b/stubs/EntityRepository.phpstub
@@ -73,12 +73,4 @@ class EntityRepository implements ObjectRepository, Selectable
     public function matching(Criteria $criteria)
     {
     }
-
-    /**
-     * @return QueryBuilder<T>
-     */
-    public function createQueryBuilder($alias, $indexBy = null): QueryBuilder
-    {
-        
-    }
 }
diff --git a/stubs/Query.phpstub b/stubs/Query.phpstub
index ae56fd2..c8b2757 100644
--- a/stubs/Query.phpstub
+++ b/stubs/Query.phpstub
@@ -8,7 +8,7 @@ namespace Doctrine\ORM;
 class Query extends AbstractQuery
 {
     /**
-     * @param int $hydrationMode
+     * @param int|string $hydrationMode
      *
      * @psalm-return (
      *    $hydrationMode is self::HYDRATE_OBJECT
diff --git a/stubs/QueryBuilder.phpstub b/stubs/QueryBuilder.phpstub
index c7bfae2..b704b2a 100644
--- a/stubs/QueryBuilder.phpstub
+++ b/stubs/QueryBuilder.phpstub
@@ -7,7 +7,6 @@ use Doctrine\ORM\Query\Expr;
 /**
  * @psalm-type _WhereExpr=Expr\Base|Expr\Comparison|Expr\Func|string
  * @psalm-type _SelectExpr=Expr\Func|string
- * @template T
  */
 class QueryBuilder
 {
@@ -93,7 +92,7 @@ class QueryBuilder
 
 
     /**
-     * @return Query<T>
+     * @return Query<mixed>
      */
     public function getQuery(): Query
     {