diff --git a/enigma/src/main/java/org/quiltmc/enigma/api/translation/mapping/serde/enigma/EnigmaMappingsReader.java b/enigma/src/main/java/org/quiltmc/enigma/api/translation/mapping/serde/enigma/EnigmaMappingsReader.java index 48ce60c1a..275522d8a 100644 --- a/enigma/src/main/java/org/quiltmc/enigma/api/translation/mapping/serde/enigma/EnigmaMappingsReader.java +++ b/enigma/src/main/java/org/quiltmc/enigma/api/translation/mapping/serde/enigma/EnigmaMappingsReader.java @@ -149,8 +149,7 @@ private static void cleanMappingStack(int indentation, Deque pair = mappingStack.pop(); if (pair.getMapping() != null) { - EntryMapping mapping = pair.getMapping().bake(); - mappings.insert(pair.getEntry(), mapping.equals(EntryMapping.DEFAULT) ? null : mapping); + mappings.insert(pair.getEntry(), pair.getMapping().bake()); } } } diff --git a/enigma/src/main/java/org/quiltmc/enigma/api/translation/mapping/tree/MergedEntryMappingTree.java b/enigma/src/main/java/org/quiltmc/enigma/api/translation/mapping/tree/MergedEntryMappingTree.java index 78e1d66b7..8ed818b42 100644 --- a/enigma/src/main/java/org/quiltmc/enigma/api/translation/mapping/tree/MergedEntryMappingTree.java +++ b/enigma/src/main/java/org/quiltmc/enigma/api/translation/mapping/tree/MergedEntryMappingTree.java @@ -42,7 +42,7 @@ public EntryMapping remove(Entry entry) { @Override public EntryMapping get(Entry entry) { EntryMapping main = this.mainTree.get(entry); - if (main == null) { + if (main == null || main.equals(EntryMapping.DEFAULT)) { return this.secondaryTree.get(entry); }