X-Git-Url: https://git.pterodactylus.net/?p=Sone.git;a=blobdiff_plain;f=src%2Ftest%2Fjava%2Fnet%2Fpterodactylus%2Fsone%2Fcore%2FSoneParserTest.java;h=3fb2a34496eebe3e94d1f3b4e3d3bca4dec6c11d;hp=6986e361934b90c4e27eae067498b2159300421a;hb=00a434a23c9ea1e57c63d8a3c0fc4b09277af431;hpb=50999afc14ced32f8f15b7512a896bbac1e61562

diff --git a/src/test/java/net/pterodactylus/sone/core/SoneParserTest.java b/src/test/java/net/pterodactylus/sone/core/SoneParserTest.java
index 6986e36..3fb2a34 100644
--- a/src/test/java/net/pterodactylus/sone/core/SoneParserTest.java
+++ b/src/test/java/net/pterodactylus/sone/core/SoneParserTest.java
@@ -345,11 +345,11 @@ public class SoneParserTest {
 
 	@Before
 	public void setupAlbums() {
-		when(core.getAlbum(anyString())).thenAnswer(new Answer<Album>() {
+		when(core.getAlbum(anyString())).thenAnswer(new Answer<Optional<Album>>() {
 			@Override
-			public Album answer(InvocationOnMock invocation)
+			public Optional<Album> answer(InvocationOnMock invocation)
 			throws Throwable {
-				return albums.get(invocation.getArguments()[0]);
+				return Optional.fromNullable(albums.get(invocation.getArguments()[0]));
 			}
 		});
 	}