diff --git a/relation_select.module b/relation_select.module
index aff6773..29cb60f 100644
--- a/relation_select.module
+++ b/relation_select.module
@@ -523,10 +523,10 @@ function _relation_select_field_widget_form(&$form, &$form_state, $field, $insta
   //
   // $items[$delta] starts as array on Edit page (Why?)
   // $items[$delta] becomes object on Preview page (Why?)
-  if (isset($items[$delta]) && is_object($items[$delta])) {
+  if (isset($items[$delta]->rid)) {
     $relation_id = $items[$delta]->rid;
   }
-  elseif (isset($items[$delta]) && is_array($items[$delta])) {
+  elseif (isset($items[$delta]['relation_id'])) {
     $relation_id = $items[$delta]['relation_id'];
   }
   else {