2 files changed, 4 insertions, 23 deletions
diff --git a/views/default/object/album/gallery.php b/views/default/object/album/gallery.php
index ee22f0436..53194b53c 100644
--- a/views/default/object/album/gallery.php
+++ b/views/default/object/album/gallery.php
@@ -10,22 +10,13 @@
 
 $album = elgg_extract('entity', $vars);
 
-$album_cover = elgg_view('output/img', array(
-	'src' => $album->getCoverImageURL(),
-	'alt' => $album->getTitle(),
-	'class' => 'elgg-photo',
-));
+$album_cover = elgg_view_entity_icon($album, 'small');
 
 $header = elgg_view('output/url', array(
 	'text' => $album->getTitle(),
 	'href' => $album->getURL(),
-));
-
-$body = elgg_view('output/url', array(
-	'text' => $album_cover,
-	'href' => $album->getURL(),
-	'encode_text' => false,
 	'is_trusted' => true,
+	'class' => 'tidypics-heading',
 ));
 
 $footer = elgg_view('output/url', array(
@@ -38,4 +29,4 @@ $footer .= '<div class="elgg-subtext">' . elgg_echo('album:num', array($album->g
 $params = array(
 	'footer' => $footer,
 );
-echo elgg_view_module('tidypics-album', $header, $body, $params);
+echo elgg_view_module('tidypics-album', $header, $album_cover, $params);
diff --git a/views/default/object/album/list.php b/views/default/object/album/list.php
index ccbd01cd1..e576c92f1 100644
--- a/views/default/object/album/list.php
+++ b/views/default/object/album/list.php
@@ -37,16 +37,6 @@ $params = array(
 $params = $params + $vars;
 $summary = elgg_view('object/elements/summary', $params);
 
-$cover = elgg_view('output/img', array(
-	'src' => $album->getCoverImageURL('thumb'),
-	'alt' => $album->getTitle(),
-	'class' => 'elgg-photo',
-));
-$icon = elgg_view('output/url', array(
-	'text' => $cover,
-	'href' => $album->getURL(),
-	'encode_text' => false,
-	'is_trusted' => true,
-));
+$icon = elgg_view_entity_icon($album, 'tiny');
 
 echo $header = elgg_view_image_block($icon, $summary);