diff --git a/clip.module b/clip.module
index 4c3802f..723dd5f 100644
--- a/clip.module
+++ b/clip.module
@@ -33,9 +33,11 @@ function clip_help($route_name, RouteMatchInterface $route_match) {
 /**
  * {@inheritdoc}
  */
-function clip_entity_type_alter(array &$entity_types) {
-  /** @var \Drupal\Core\Entity\EntityTypeInterface[] $entity_types */
-  $classes = $entity_types['media']->getHandlerClasses();
-  $classes['form']['edit'] = 'Drupal\clip\Form\EditMediaForm';
-  $entity_types['media']->setHandlerClass('form', $classes['form']);
-}
+  function clip_entity_type_alter(array &$entity_types) {
+    if (isset($entity_types['media'])) {
+    /** @var \Drupal\Core\Entity\EntityTypeInterface[] $entity_types */
+    $classes = $entity_types['media']->getHandlerClasses();
+    $classes['form']['edit'] = 'Drupal\clip\Form\EditMediaForm';
+    $entity_types['media']->setHandlerClass('form', $classes['form']);
+    }
+  }