summaryrefslogtreecommitdiff
diff options
context:
space:
mode:
-rw-r--r--enigma/src/main/java/cuchaz/enigma/source/SourceIndex.java7
1 files changed, 5 insertions, 2 deletions
diff --git a/enigma/src/main/java/cuchaz/enigma/source/SourceIndex.java b/enigma/src/main/java/cuchaz/enigma/source/SourceIndex.java
index a38c6f55..fec20e25 100644
--- a/enigma/src/main/java/cuchaz/enigma/source/SourceIndex.java
+++ b/enigma/src/main/java/cuchaz/enigma/source/SourceIndex.java
@@ -141,8 +141,11 @@ public class SourceIndex {
141 tokenToReference.replace(token, resolvedReference); 141 tokenToReference.replace(token, resolvedReference);
142 142
143 Collection<Token> tokens = referenceToTokens.remove(reference); 143 Collection<Token> tokens = referenceToTokens.remove(reference);
144 referenceToTokens.computeIfAbsent(resolvedReference, key -> new ArrayList<>()) 144
145 .addAll(tokens); 145 if (tokens != null) {
146 referenceToTokens.computeIfAbsent(resolvedReference, key -> new ArrayList<>())
147 .addAll(tokens);
148 }
146 } 149 }
147 } 150 }
148 151