diff --git a/inline_entity_form.module b/inline_entity_form.module
index a0d2332..861ea17 100644
--- a/inline_entity_form.module
+++ b/inline_entity_form.module
@@ -800,7 +800,7 @@ function inline_entity_form_element_validate($element, &$form_state, $form) {
  * Implements hook_field_attach_submit().
  */
 function inline_entity_form_field_attach_submit($entity_type, $entity, $form, &$form_state) {
-  $bundle_name = field_extract_bundle($entity_type, $entity);
+  list(,,$bundle_name) = entity_extract_ids($entity_type, $entity);
 
   foreach (field_info_instances($entity_type, $bundle_name) as $instance_name => $instance) {
     if (isset($instance['widget']) && $instance['widget']['type'] == 'inline_entity_form') {