diff --git a/prev_next_views.module b/prev_next_views.module
index 7dc86f7..631515e 100644
--- a/prev_next_views.module
+++ b/prev_next_views.module
@@ -93,13 +93,13 @@ function prev_next_views_views_api() {
 /**
  * Implements hook_views_pre_execute().
  */
-function prev_next_views_views_pre_execute(&$view) {
+function prev_next_views_views_post_execute(&$view) {
   if (in_array($view->name, variable_get('prev_next_views_enabled', array()))) {
-    $result = $view->build_info['query']->execute()->fetchAllAssoc('nid');
+    $result = $view->result;
 
     $nids = array();
     foreach ($result as $item) {
-      $nids[] = $item->nid;
+      $nids[] = $item->entity;
     }
 
     $_SESSION['prev_next_views'] = $nids;