diff --git a/src/Form/RestUIForm.php b/src/Form/RestUIForm.php
index dee482a..01557ae 100644
--- a/src/Form/RestUIForm.php
+++ b/src/Form/RestUIForm.php
@@ -13,7 +13,6 @@ use Drupal\Core\Config\ConfigFactoryInterface;
 use Drupal\Core\Config\Context\ContextInterface;
 use Drupal\Core\Extension\ModuleHandler;
 use Drupal\Core\Url;
-use Symfony\Component\HttpFoundation\RedirectResponse;
 use Symfony\Component\DependencyInjection\ContainerInterface;
 use Symfony\Component\HttpKernel\Exception\NotFoundHttpException;
 use Drupal\rest\Plugin\Type\ResourcePluginManager;
diff --git a/src/RestUIManager.php b/src/RestUIManager.php
index f2e4e37..4d28a1d 100644
--- a/src/RestUIManager.php
+++ b/src/RestUIManager.php
@@ -1,4 +1,5 @@
 <?php
+
 /**
  * @file
  * Contains \Drupal\restui\RestUIManager.
@@ -6,9 +7,7 @@
 
 namespace Drupal\restui;
 
-use Drupal\Core\Entity\EntityManager;
-use Drupal\Core\Entity\EntityManagerInterface;
-use Symfony\Component\DependencyInjection\ContainerInterface;
+use Drupal\Core\Entity\EntityTypeManagerInterface;
 
 /**
  * Rest UI Manager Service.
@@ -31,8 +30,11 @@ class RestUIManager {
 
   /**
    * Constructs a RestUIManager object.
+   *
+   * @param EntityTypeManagerInterface $entityManager
+   *   An entity type manager interface.
    */
-  public function __construct(EntityManagerInterface $entityManager) {
+  public function __construct(EntityTypeManagerInterface $entityManager) {
     $this->entityManager = $entityManager;
   }