diff --git a/modules/entity_form/entity_browser_entity_form.module b/modules/entity_form/entity_browser_entity_form.module
index 90b4350..61f2fcf 100644
--- a/modules/entity_form/entity_browser_entity_form.module
+++ b/modules/entity_form/entity_browser_entity_form.module
@@ -39,7 +39,11 @@ function entity_browser_entity_form_inline_entity_form_reference_form_alter(&$re
   }
 
   // Fetch the number of currently selected entities, if any.
-  $count_existing_selection = !empty($form_state->getUserInput()[$reference_form['#parents'][0]]['entities']) ? count($form_state->getUserInput()[$reference_form['#parents'][0]]['entities']) : 0;
+  $count_existing_selection = count($form_state->get([
+    'inline_entity_form',
+    $reference_form['#ief_id'],
+    'entities',
+  ]));
 
   unset($reference_form['entity_id']);
   $reference_form['entity_browser'] = [