diff --git a/CHANGELOG.md b/CHANGELOG.md
index 12a6bb3..f421484 100644
--- a/CHANGELOG.md
+++ b/CHANGELOG.md
@@ -11,6 +11,7 @@ before starting to add changes. Use example [placed in the end of the page](#exa
 
 ## [Unreleased]
 
+- Added missing return type.
 - Added new webform submission storage class to handle enable both encryption
   and revision at the same time.
 
diff --git a/modules/os2forms_webform_list/src/CustomWebformEntityListBuilder.php b/modules/os2forms_webform_list/src/CustomWebformEntityListBuilder.php
index 94e64fb..1eb2731 100644
--- a/modules/os2forms_webform_list/src/CustomWebformEntityListBuilder.php
+++ b/modules/os2forms_webform_list/src/CustomWebformEntityListBuilder.php
@@ -2,6 +2,7 @@
 
 namespace Drupal\os2forms_webform_list;
 
+use Drupal\Core\Entity\Query\QueryInterface;
 use Drupal\webform\WebformEntityListBuilder;
 
 /**
@@ -27,7 +28,7 @@ class CustomWebformEntityListBuilder extends WebformEntityListBuilder {
    * @return \Drupal\Core\Entity\Query\QueryInterface
    *   An entity query.
    */
-  protected function getQuery($keys = '', $category = '', $state = '') {
+  protected function getQuery($keys = '', $category = '', $state = ''): QueryInterface {
     $query = parent::getQuery($keys, $category, $state);
 
     // Setup a required condition for the list builder to respect webform update