diff --git a/theme/theme.inc b/theme/theme.inc
index 27470c3..84c5bf6 100644
--- a/theme/theme.inc
+++ b/theme/theme.inc
@@ -35,8 +35,8 @@ function relation_select_preprocess_views_view_relation_select(&$variables){
     else {
       $entity_id = $result[$num]->{$view->query->base_field};
       $entity_type = relation_select_entity_get_type_from_view($view);
-      $entity = entity_load_single($entity_type, $entity_id);
-      $bundle = relation_select_entity_get_bundle($entity_type, $entity);
+      $entity = entity_load($entity_type, array($entity_id));
+      $bundle = relation_select_entity_get_bundle($entity_type, reset($entity));
     }
     $variables['row_entity'][$num] = array(
       'entity_type' => $entity_type,