X-Git-Url: https://git.pterodactylus.net/?a=blobdiff_plain;f=src%2Fmain%2Fjava%2Fnet%2Fpterodactylus%2Fsone%2Ftemplate%2FAlbumAccessor.java;h=51cd974d7388ed266da648d099ff82f516e9288c;hb=28b2e2b0d2ff2320f5b6a5ac36a858a604c7c7ba;hp=13b4d65c766b6a6828c9ce36636f9e8446877374;hpb=7017646bf42cb265b6df539bb6def40b91d2f968;p=Sone.git

diff --git a/src/main/java/net/pterodactylus/sone/template/AlbumAccessor.java b/src/main/java/net/pterodactylus/sone/template/AlbumAccessor.java
index 13b4d65..51cd974 100644
--- a/src/main/java/net/pterodactylus/sone/template/AlbumAccessor.java
+++ b/src/main/java/net/pterodactylus/sone/template/AlbumAccessor.java
@@ -43,12 +43,14 @@ public class AlbumAccessor extends ReflectionAccessor {
 		if ("backlinks".equals(member)) {
 			List<Link> backlinks = new ArrayList<Link>();
 			Album currentAlbum = album;
-			while (!currentAlbum.equals(album.getSone().getRootAlbum())) {
+			while (!currentAlbum.isRoot()) {
 				backlinks.add(0, new Link("imageBrowser.html?album=" + currentAlbum.getId(), currentAlbum.getTitle()));
 				currentAlbum = currentAlbum.getParent();
 			}
 			backlinks.add(0, new Link("imageBrowser.html?sone=" + album.getSone().getId(), SoneAccessor.getNiceName(album.getSone())));
 			return backlinks;
+		} else if ("albumImage".equals(member)) {
+			return album.getAlbumImage().orNull();
 		}
 		return super.get(templateContext, object, member);
 	}