X-Git-Url: https://git.pterodactylus.net/?a=blobdiff_plain;f=src%2Fmain%2Fjava%2Fnet%2Fpterodactylus%2Fsone%2Fcore%2FCore.java;h=89de6fa1b92f9d617980b1495c1ec0ad98be2e32;hb=2b400308146190305dde0778627cd138fe2a0d0b;hp=a67177bafe8968c3fd516a7cc6e5ba63f48430db;hpb=a23c4f218c3adf236d89d5927cae37d6e6e4feda;p=Sone.git

diff --git a/src/main/java/net/pterodactylus/sone/core/Core.java b/src/main/java/net/pterodactylus/sone/core/Core.java
index a67177b..89de6fa 100644
--- a/src/main/java/net/pterodactylus/sone/core/Core.java
+++ b/src/main/java/net/pterodactylus/sone/core/Core.java
@@ -1548,16 +1548,7 @@ public class Core implements IdentityListener, UpdateListener, ImageInsertListen
 			configuration.getStringValue(sonePrefix + "/Friends/" + friendCounter + "/ID").setValue(null);
 
 			/* save albums. first, collect in a flat structure, top-level first. */
-			List<Album> albums = new ArrayList<Album>();
-			albums.addAll(sone.getAlbums());
-			int lastAlbumIndex = 0;
-			while (lastAlbumIndex < albums.size()) {
-				int previousAlbumCount = albums.size();
-				for (Album album : new ArrayList<Album>(albums.subList(lastAlbumIndex, albums.size()))) {
-					albums.addAll(album.getAlbums());
-				}
-				lastAlbumIndex = previousAlbumCount;
-			}
+			List<Album> albums = Sone.flattenAlbums(sone.getAlbums());
 
 			int albumCounter = 0;
 			for (Album album : albums) {