diff --git a/src/ConfigLister.php b/src/ConfigLister.php
index 06219cc..a2823df 100644
--- a/src/ConfigLister.php
+++ b/src/ConfigLister.php
@@ -34,7 +34,7 @@ class ConfigLister implements ConfigListInterface {
   /**
    * The entity manager.
    *
-   * @var \Drupal\Core\Entity\EntityManagerInterface
+   * @var \Drupal\Core\Entity\EntityTypeManagerInterface
    */
   protected $entityManager;
 
diff --git a/src/ConfigReverter.php b/src/ConfigReverter.php
index e47103c..daec851 100644
--- a/src/ConfigReverter.php
+++ b/src/ConfigReverter.php
@@ -19,7 +19,7 @@ class ConfigReverter implements ConfigRevertInterface, ConfigDeleteInterface {
   /**
    * The entity manager.
    *
-   * @var \Drupal\Core\Entity\EntityManagerInterface
+   * @var \Drupal\Core\Entity\EntityTypeManagerInterface
    */
   protected $entityManager;