Retrieve insert token by image ID.

diff --git a/src/main/java/net/pterodactylus/sone/core/ImageInserter.java b/src/main/java/net/pterodactylus/sone/core/ImageInserter.java
index 930fc48..b4421ce 100644 (file)
--- a/src/main/java/net/pterodactylus/sone/core/ImageInserter.java
+++ b/src/main/java/net/pterodactylus/sone/core/ImageInserter.java
@@ -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;
                }