diff --git a/inline_entity_form.module b/inline_entity_form.module
index 983cf6e..5fab411 100644
--- a/inline_entity_form.module
+++ b/inline_entity_form.module
@@ -268,11 +268,12 @@ function inline_entity_form_close_all_forms($elements, FormStateInterface $form_
     // Close the main form.
     $form_state->set(['inline_entity_form', $ief_id, 'form'], NULL);
     // Close the row forms.
-    $entities = $form_state->get(['inline_entity_form', $ief_id, 'entities']);
-    foreach ($entities as $key => $value) {
-      $entities[$key]['form'] = NULL;
+    if ($entities = $form_state->get(['inline_entity_form', $ief_id, 'entities'])) {
+      foreach ($entities as $key => $value) {
+        $entities[$key]['form'] = NULL;
+      }
+      $form_state->set(['inline_entity_form', $ief_id, 'entities'], $entities);
     }
-    $form_state->set(['inline_entity_form', $ief_id, 'entities'], $entities);
   }
 }