diff --git a/usagov_benefit_finder/modules/usagov_benefit_finder_content/usagov_benefit_finder_content.module b/usagov_benefit_finder/modules/usagov_benefit_finder_content/usagov_benefit_finder_content.module index 557a9013..95bb6514 100644 --- a/usagov_benefit_finder/modules/usagov_benefit_finder_content/usagov_benefit_finder_content.module +++ b/usagov_benefit_finder/modules/usagov_benefit_finder_content/usagov_benefit_finder_content.module @@ -6,7 +6,9 @@ */ use Drupal\Core\Database\Database; +use Drupal\Core\Entity\EntityForm; use Drupal\Core\Form\FormStateInterface; +use Drupal\node\NodeInterface; use Drupal\node\Entity\Node; /** @@ -482,8 +484,16 @@ function usagov_benefit_finder_content_form_node_bears_life_event_form_edit_form function _usagov_benefit_finder_content_check_life_event_form_criteria_depth(array &$form, FormStateInterface $form_state) { $error_flag = FALSE; $criteria_depths = []; + $life_event_form_node = NULL; + + $form_object = $form_state->getFormObject(); + if ($form_object instanceof EntityForm) { + $entity = $form_object->getEntity(); + if ($entity instanceof NodeInterface) { + $life_event_form_node = $entity; + } + } - $life_event_form_node = $form_state->getFormObject()->getEntity(); $sections = $life_event_form_node->get('field_b_sections_elg_criteria')->referencedEntities(); foreach ($sections as $section) {