X-Git-Url: https://git.pterodactylus.net/?p=Sone.git;a=blobdiff_plain;f=src%2Fmain%2Fjava%2Fnet%2Fpterodactylus%2Fsone%2Fdatabase%2Fmemory%2FMemoryDatabase.java;h=7ad40e0e36799f943975a610efc5ce5e51fb8700;hp=128a4bd8b716e554d96433d0edc7c1ebd0f3b01b;hb=3049b7e396ce20b5b6600ae2791334a6babb414e;hpb=5bcc4f8e8d920408cf0c940ba67f6a1ebf9c4228

diff --git a/src/main/java/net/pterodactylus/sone/database/memory/MemoryDatabase.java b/src/main/java/net/pterodactylus/sone/database/memory/MemoryDatabase.java
index 128a4bd..7ad40e0 100644
--- a/src/main/java/net/pterodactylus/sone/database/memory/MemoryDatabase.java
+++ b/src/main/java/net/pterodactylus/sone/database/memory/MemoryDatabase.java
@@ -173,14 +173,14 @@ public class MemoryDatabase extends AbstractService implements Database {
 	}
 
 	@Override
-	public Sone registerLocalSone(OwnIdentity ownIdentity) {
-		final Sone localSone = loadLocalSone(ownIdentity);
+	public LocalSone registerLocalSone(OwnIdentity ownIdentity) {
+		final LocalSone localSone = loadLocalSone(ownIdentity);
 		localSones.add(ownIdentity.getId());
 		return localSone;
 	}
 
-	private Sone loadLocalSone(OwnIdentity ownIdentity) {
-		Sone localSone = newSoneBuilder().local().from(ownIdentity).build();
+	private LocalSone loadLocalSone(OwnIdentity ownIdentity) {
+		LocalSone localSone = (LocalSone) newSoneBuilder().local().from(ownIdentity).build();
 		localSone.setLatestEdition(
 				Optional.fromNullable(
 						Longs.tryParse(ownIdentity.getProperty(LATEST_EDITION_PROPERTY)))
@@ -192,7 +192,7 @@ public class MemoryDatabase extends AbstractService implements Database {
 		return localSone;
 	}
 
-	public void loadSone(Sone sone) {
+	public void loadSone(LocalSone sone) {
 		long soneTime = configurationLoader.getLocalSoneTime(sone.getId());
 		if (soneTime == -1) {
 			return;