diff --git a/src/Interface/QueryBuilderGeneratedInterface.php b/src/Interface/QueryBuilderGeneratedInterface.php
new file mode 100644
index 0000000..20c1dcd
--- /dev/null
+++ b/src/Interface/QueryBuilderGeneratedInterface.php
@@ -0,0 +1,41 @@
+<?php
+
+namespace Tbn\QueryBuilderRepositoryGeneratorBundle\Interface;
+
+use Doctrine\ORM\QueryBuilder;
+
+interface QueryBuilderGeneratedInterface
+{
+    public function findOne(
+        mixed $id,
+        bool $useQueryCache = false,
+        ?string $cacheId = null,
+        array $resultCacheTags = [],
+    ): object;
+
+    public static function filterIn(
+        QueryBuilder $qb,
+        string $columnName,
+        string $operator,
+        array $values,
+        ?string $entityName = null,
+    ): QueryBuilder;
+
+    public static function filterBy(
+        QueryBuilder $qb,
+        string $columnName,
+        string $operator,
+        mixed $value,
+        ?string $entityName = null,
+    ): QueryBuilder;
+
+    public function getNewQueryBuilder(): QueryBuilder;
+
+    public function getQueryBuilderCount(
+        QueryBuilder $qb,
+        string $entityName,
+        bool $useQueryCache = false,
+        ?string $cacheId = null,
+        array $resultCacheTags = [],
+    ): int;
+}
diff --git a/src/Resources/views/Generator/TopRepositoryTemplate.html.twig b/src/Resources/views/Generator/TopRepositoryTemplate.html.twig
index 1ae7337..783e6da 100644
--- a/src/Resources/views/Generator/TopRepositoryTemplate.html.twig
+++ b/src/Resources/views/Generator/TopRepositoryTemplate.html.twig
@@ -18,13 +18,14 @@ use Symfony\Contracts\Cache\ItemInterface;
 {% if useInterfaces %}
 use Tbn\QueryBuilderRepositoryGeneratorBundle\Interface\FilterByInterface;
 use Tbn\QueryBuilderRepositoryGeneratorBundle\Interface\FilterInInterface;
+use Tbn\QueryBuilderRepositoryGeneratorBundle\Interface\QueryBuilderGeneratedInterface;
 {% endif %}
 
 /**
  * Generated class for filter query builders
  *
  */
-class {{ entityClassname }}Base extends {{ extendClass }}{% if useInterfaces %} implements FilterByInterface, FilterInInterface{% endif %}
+class {{ entityClassname }}Base extends {{ extendClass }}{% if useInterfaces %} implements FilterByInterface, FilterInInterface, QueryBuilderGeneratedInterface{% endif %}
 {
     const NAME = '{{ entityDql }}';
     protected static $parameterIndex = 0;