fix(modlauncher): add custom ois classes to transformer targets

dogboy21 · Aug 27, 2023 · 9b716f4 · 9b716f4
1 parent 332b69e
commit 9b716f4
Showing 1 changed file with 3 additions and 1 deletion.
diff --git a/modlauncher/src/main/java/io/dogboy/serializationisbad/modlauncher/SIBTransformer.java b/modlauncher/src/main/java/io/dogboy/serializationisbad/modlauncher/SIBTransformer.java
@@ -11,6 +11,7 @@
 
 import java.util.Set;
 import java.util.stream.Collectors;
+import java.util.stream.Stream;
 
 public class SIBTransformer implements ITransformer<ClassNode> {
     private final PatchModule patchModule;
@@ -36,7 +37,8 @@ public TransformerVoteResult castVote(ITransformerVotingContext context) {
 
     @Override
     public Set<Target> targets() {
-        return this.patchModule.getClassesToPatch().stream()
+        return Stream.concat(this.patchModule.getClassesToPatch().stream(),
+                        this.patchModule.getCustomOISClasses().stream())
                 .map(Target::targetClass)
                 .collect(Collectors.toSet());
     }