diff options
| -rw-r--r-- | enigma/src/main/java/cuchaz/enigma/source/SourceIndex.java | 7 |
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 | ||