diff --git a/viewreference.module b/viewreference.module
index d9a8a69..323a1ed 100755
--- a/viewreference.module
+++ b/viewreference.module
@@ -475,6 +475,7 @@ function viewreference_get_view($element, $params = array()) {
         }
       }
       if ((in_array('title', $params) && empty($view['title'])) || (in_array('link', $params) && empty($view['link']))) {
+        $view['display_key'] = $view['display'];
         $view['title'] = theme('viewreference_display_title', $view);
         if (in_array('link', $params) && empty($view['link'])) {
           $disabled = isset($view['view']->disabled) ? $view['view']->disabled:  FALSE;