diff --git a/nodereference_url.module b/nodereference_url.module
index 5268fad..372b576 100644
--- nodereference_url.module
+++ nodereference_url.module
@@ -392,7 +392,7 @@ function nodereference_url_widget(&$form, &$form_state, $field, $items, $delta =
     }
   }
 
-  if (isset($referenced_nid) && (empty($field['widget']['edit_fallback']))) {
+  if (isset($referenced_nid) && (empty($element['#type']))) {
     $element[0]['nid'] = array(
       '#title' => $field['widget']['label'],
       '#type' => 'nodereference_url',