diff --git a/templates/bake/element/table_methods.twig b/templates/bake/element/table_methods.twig
index 7f8afc4..27d05cb 100644
--- a/templates/bake/element/table_methods.twig
+++ b/templates/bake/element/table_methods.twig
@@ -3,7 +3,7 @@
      *
      * @param \Authorization\IdentityInterface $user The user.
      * @param \Cake\ORM\Query\SelectQuery $query The query to apply authorization conditions to.
-     * @return bool
+     * @return \Cake\ORM\Query\SelectQuery
      */
     public function scopeIndex(IdentityInterface $user, SelectQuery $query): SelectQuery
     {
diff --git a/tests/comparisons/BookmarksTablePolicy.php b/tests/comparisons/BookmarksTablePolicy.php
index f2fdb8a..2d11848 100644
--- a/tests/comparisons/BookmarksTablePolicy.php
+++ b/tests/comparisons/BookmarksTablePolicy.php
@@ -16,7 +16,7 @@ class BookmarksTablePolicy
      *
      * @param \Authorization\IdentityInterface $user The user.
      * @param \Cake\ORM\Query\SelectQuery $query The query to apply authorization conditions to.
-     * @return bool
+     * @return \Cake\ORM\Query\SelectQuery
      */
     public function scopeIndex(IdentityInterface $user, SelectQuery $query): SelectQuery
     {
diff --git a/tests/comparisons/TestPluginUsersTablePolicy.php b/tests/comparisons/TestPluginUsersTablePolicy.php
index 7c4789d..7f632f3 100644
--- a/tests/comparisons/TestPluginUsersTablePolicy.php
+++ b/tests/comparisons/TestPluginUsersTablePolicy.php
@@ -16,7 +16,7 @@ class UsersTablePolicy
      *
      * @param \Authorization\IdentityInterface $user The user.
      * @param \Cake\ORM\Query\SelectQuery $query The query to apply authorization conditions to.
-     * @return bool
+     * @return \Cake\ORM\Query\SelectQuery
      */
     public function scopeIndex(IdentityInterface $user, SelectQuery $query): SelectQuery
     {