diff --git a/webform.module b/webform.module
index 612adb0..5b5fd01 100644
--- a/webform.module
+++ b/webform.module
@@ -3399,10 +3399,13 @@ function webform_views_api() {
 function webform_field_extra_fields() {
   $extra = array();
   foreach (webform_variable_get('webform_node_types') as $type) {
-    $extra['node'][$type]['webform'] = array(
-      'label' => t('Webform'),
-      'description' => t('Webform client form.'),
-      'weight' => 10,
+    'display' => array(
+      $type  => array(
+        'label' => t('Webform'),
+        'description' => t('Webform client form.'),
+        'weight' => 10,
+        ),
+      ),
     );
   }
   return $extra;
