Set Sone in album builder, remote getOrCreate method from core.

[Sone.git] / src / main / java / net / pterodactylus / sone / data / impl / AlbumBuilderImpl.java

diff --git a/src/main/java/net/pterodactylus/sone/data/impl/AlbumBuilderImpl.java b/src/main/java/net/pterodactylus/sone/data/impl/AlbumBuilderImpl.java
index 3403a62..785a4af 100644 (file)
--- a/src/main/java/net/pterodactylus/sone/data/impl/AlbumBuilderImpl.java
+++ b/src/main/java/net/pterodactylus/sone/data/impl/AlbumBuilderImpl.java
@@ -31,7 +31,7 @@ public class AlbumBuilderImpl extends AbstractAlbumBuilder {
        @Override
        public Album build() throws IllegalStateException {
                validate();
-               return randomId ? new AlbumImpl() : new AlbumImpl(id);
+               return randomId ? new AlbumImpl(sone) : new AlbumImpl(sone, id);
        }
 
 }