Silvi Sanghavi
--- a/entity_usage.module
+++ b/entity_usage.module
 */
 function entity_usage_form_alter(&$form, FormStateInterface $form_state, $form_id) {
  $form_object = $form_state->getFormObject();
-  if (!method_exists($form_object, 'getEntity')) {
+  if (!$form_object instanceof \Drupal\Core\Entity\EntityFormInterface) {
-  if (!$form_object instanceof \Drupal\Core\Entity\EntityFormInterface) {
+  if (!$form_object instanceof EntityFormInterface) {
--- a/entity_usage.module
+++ b/entity_usage.module
  if (!in_array($entity_type_id, $edit_entity_types) && !in_array($entity_type_id, $delete_entity_types)) {
    return;
  }
  $is_edit_form = method_exists($form_object, 'getOperation') && $form_object->getOperation() === 'edit' && in_array($entity_type_id, $edit_entity_types);
-  $is_edit_form = method_exists($form_object, 'getOperation') && $form_object->getOperation() === 'edit' && in_array($entity_type_id, $edit_entity_types);
+  $is_edit_form = $form_object->getOperation() === 'edit' && in_array($entity_type_id, $edit_entity_types);