diff --git a/src/FlagListsFlagListBuilder.php b/src/FlagListsFlagListBuilder.php
index 6773759..f6a1242 100644
--- a/src/FlagListsFlagListBuilder.php
+++ b/src/FlagListsFlagListBuilder.php
@@ -3,7 +3,7 @@
 namespace Drupal\flag_lists;
 
 use Drupal\Core\Config\Entity\ConfigEntityType;
-use Drupal\Core\Config\Entity\ConfigEntityStorage;
+use Drupal\Core\Entity\EntityStorageInterface;
 use Drupal\Core\Config\ConfigFactoryInterface;
 use Drupal\Core\Entity\EntityInterface;
 use Drupal\Core\Entity\EntityTypeInterface;
@@ -55,7 +55,7 @@ class FlagListsFlagListBuilder extends FlagListBuilder {
    *
    * @param Drupal\Core\Config\Entity\ConfigEntityType $entity_type
    *   The entity type involved.
-   * @param Drupal\Core\Config\Entity\ConfigEntityStorage $storage
+   * @param Drupal\Core\Entity\EntityStorageInterface $storage
    *   The entity storage.
    * @param Drupal\Core\Config\ConfigFactoryInterface $config_factory
    *   The config factory.
@@ -63,7 +63,7 @@ class FlagListsFlagListBuilder extends FlagListBuilder {
    *   The Flag Lists Service.
    */
   public function __construct(ConfigEntityType $entity_type,
-    ConfigEntityStorage $storage,
+    EntityStorageInterface $storage,
     ConfigFactoryInterface $config_factory,
     FlagListsService $flag_lists_service) {
     parent::__construct($entity_type, $storage);