diff --git a/themes/media_flickr.theme.inc b/themes/media_flickr.theme.inc
index f8c0225..7c135b9 100644
--- a/themes/media_flickr.theme.inc
+++ b/themes/media_flickr.theme.inc
@@ -19,7 +19,11 @@ function theme_media_flickr_styles($variables) {
 
 function media_flickr_preprocess_media_flickr_photoset(&$variables) {
   // Build the URL for display.
-  $uri = $variables['uri'];
+  if ($variables['uri']) {
+    $uri = $variables['uri'];
+  } else {
+    $uri = $variables["object"]->file->uri;
+  }
   $wrapper = file_stream_wrapper_get_instance_by_uri($uri);
   $parts = $wrapper->get_parameters();
   if (isset($parts['search'])) {