diff --git a/plugins/content_types/form/entity_form_field.inc b/plugins/content_types/form/entity_form_field.inc
index 3905a31..5bc4bcf 100644
--- a/plugins/content_types/form/entity_form_field.inc
+++ b/plugins/content_types/form/entity_form_field.inc
@@ -73,12 +73,11 @@ function ctools_entity_form_field_content_type_render($subtype, $conf, $panel_ar
   }
 
   // Get a shortcut to the entity.
-  $entity = $context->data;
+  $bundle = $context->data;
   list($entity_type, $field_name) = explode(':', $subtype, 2);
 
   // Load the entity type's information for this field.
-  $ids = entity_extract_ids($entity_type, $entity);
-  $field = field_info_instance($entity_type, $field_name, $ids[2]);
+  $field = field_info_instance($entity_type, $field_name, $bundle);
 
   // Do not render if the entity type does not have this field.
   if (empty($field)) {