X-Git-Url: https://git.pterodactylus.net/?a=blobdiff_plain;f=src%2Fmain%2Fjava%2Fnet%2Fpterodactylus%2Fsone%2Fcore%2FImageInserter.java;h=b4421ce6514e2919fafdb4da1a8623958b490719;hb=cb6325fa54d93612c5aad307204d30600b27af81;hp=18819026a73a258d42ca101cb54b5f953018eece;hpb=f4d5dbe7d115fb18ec375d5e109c18359d50cbd9;p=Sone.git

diff --git a/src/main/java/net/pterodactylus/sone/core/ImageInserter.java b/src/main/java/net/pterodactylus/sone/core/ImageInserter.java
index 1881902..b4421ce 100644
--- a/src/main/java/net/pterodactylus/sone/core/ImageInserter.java
+++ b/src/main/java/net/pterodactylus/sone/core/ImageInserter.java
@@ -76,7 +76,7 @@ public class ImageInserter {
 	public void insertImage(TemporaryImage temporaryImage, Image image) {
 		Validation.begin().isNotNull("Temporary Image", temporaryImage).isNotNull("Image", image).check().isEqual("Image IDs", image.getId(), temporaryImage.getId()).check();
 		try {
-			InsertToken insertToken = new InsertToken(image);
+			InsertToken insertToken = freenetInterface.new InsertToken(image);
 			insertTokens.put(image.getId(), insertToken);
 			insertToken.addImageInsertListener(core);
 			freenetInterface.insertImage(temporaryImage, image, insertToken);
@@ -93,7 +93,7 @@ public class ImageInserter {
 	 *            The image being inserted
 	 */
 	public void cancelImageInsert(Image image) {
-		InsertToken insertToken = insertTokens.remove(image);
+		InsertToken insertToken = insertTokens.remove(image.getId());
 		if (insertToken == null) {
 			return;
 		}