diff --git a/config/schema/workbench_moderation.schema.yml b/config/schema/workbench_moderation.schema.yml
index fd5cd31..6bd4112 100644
--- a/config/schema/workbench_moderation.schema.yml
+++ b/config/schema/workbench_moderation.schema.yml
@@ -52,7 +52,6 @@ block_content.type.*.third_party.workbench_moderation:
 views.filter.latest_revision:
   type: views_filter
   label: 'Latest revision'
-  type: mapping
   mapping:
     value:
       type: string
diff --git a/src/ModerationStateTransitionStorage.php b/src/ModerationStateTransitionStorage.php
index 571b2b4..e42bf9e 100644
--- a/src/ModerationStateTransitionStorage.php
+++ b/src/ModerationStateTransitionStorage.php
@@ -4,6 +4,7 @@ namespace Drupal\workbench_moderation;
 
 
 use Drupal\Component\Uuid\UuidInterface;
+use Drupal\Core\Cache\MemoryCache\MemoryCacheInterface;
 use Drupal\Core\Config\ConfigFactoryInterface;
 use Drupal\Core\Config\Entity\ConfigEntityStorage;
 use Drupal\Core\Entity\EntityHandlerInterface;
@@ -30,15 +31,16 @@ class ModerationStateTransitionStorage extends ConfigEntityStorage implements En
       $container->get('config.factory'),
       $container->get('uuid'),
       $container->get('language_manager'),
-      $container->get('entity_type.manager')
+      $container->get('entity_type.manager'),
+      $container->get('entity.memory_cache')
     );
   }
 
   /**
    * {@inheritdoc}
    */
-  public function __construct(EntityTypeInterface $entity_type, ConfigFactoryInterface $config_factory, UuidInterface $uuid_service, LanguageManagerInterface $language_manager, EntityTypeManagerInterface $entity_type_manager) {
-    parent::__construct($entity_type, $config_factory, $uuid_service, $language_manager);
+  public function __construct(EntityTypeInterface $entity_type, ConfigFactoryInterface $config_factory, UuidInterface $uuid_service, LanguageManagerInterface $language_manager, EntityTypeManagerInterface $entity_type_manager, MemoryCacheInterface $memory_cache) {
+    parent::__construct($entity_type, $config_factory, $uuid_service, $language_manager, $memory_cache);
     $this->entityTypeManager = $entity_type_manager;
   }