diff --git a/disunity-core/src/main/java/info/ata4/junity/serialize/typetree/TypeTreeV3.java b/disunity-core/src/main/java/info/ata4/junity/serialize/typetree/TypeTreeV3.java index b5bd2f2e..c79a1e88 100644 --- a/disunity-core/src/main/java/info/ata4/junity/serialize/typetree/TypeTreeV3.java +++ b/disunity-core/src/main/java/info/ata4/junity/serialize/typetree/TypeTreeV3.java @@ -22,7 +22,7 @@ import java.util.List; import java.util.Map; import java.util.concurrent.atomic.AtomicInteger; -import java.util.function.Function; +import java.util.function.ToIntFunction; /** * @@ -167,7 +167,7 @@ private void writeNode(DataWriter out, Node node) throws IOException { Map localMap = new LinkedHashMap<>(); Map commonMap = StringTable.commonStrings(revision.major()).inverse(); - Function addStringOffset = typeName -> { + ToIntFunction addStringOffset = typeName -> { if (commonMap.containsKey(typeName)) { return commonMap.get(typeName); } else if (localMap.containsKey(typeName)) { @@ -181,8 +181,8 @@ private void writeNode(DataWriter out, Node node) throws IOException { // apply string offsets types.forEach(type -> { - type.typeOffset(addStringOffset.apply(type.typeName())); - type.nameOffset(addStringOffset.apply(type.fieldName())); + type.typeOffset(addStringOffset.applyAsInt(type.typeName())); + type.nameOffset(addStringOffset.applyAsInt(type.fieldName())); }); out.writeInt(types.size());