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

diff --git a/src/main/java/net/pterodactylus/sone/core/ImageInserter.java b/src/main/java/net/pterodactylus/sone/core/ImageInserter.java
index 1881902..4fc7e89 100644
--- a/src/main/java/net/pterodactylus/sone/core/ImageInserter.java
+++ b/src/main/java/net/pterodactylus/sone/core/ImageInserter.java
@@ -1,5 +1,5 @@
 /*
- * Sone - ImageInserter.java - Copyright © 2011 David Roden
+ * Sone - ImageInserter.java - Copyright © 2011–2012 David Roden
  *
  * This program is free software: you can redistribute it and/or modify
  * it under the terms of the GNU General Public License as published by
@@ -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;
 		}