From 1dd3394ea38b7e7feb4c6e15a440e8e50b577058 Mon Sep 17 00:00:00 2001
From: oSumAtrIX <johan.melkonyan1@web.de>
Date: Thu, 31 Mar 2022 18:56:36 +0200
Subject: [PATCH] refactor: Replacing original classes with mutated ones

---
 src/main/kotlin/app/revanced/patcher/Patcher.kt | 7 ++++---
 1 file changed, 4 insertions(+), 3 deletions(-)

diff --git a/src/main/kotlin/app/revanced/patcher/Patcher.kt b/src/main/kotlin/app/revanced/patcher/Patcher.kt
index e55ef91..7abee14 100644
--- a/src/main/kotlin/app/revanced/patcher/Patcher.kt
+++ b/src/main/kotlin/app/revanced/patcher/Patcher.kt
@@ -29,12 +29,13 @@ class Patcher(
         val newDexFile = object : DexFile {
             override fun getClasses(): Set<ClassDef> {
                 // this is a slow workaround for now
-                val mutableClassList = cache.classes.toMutableList()
+                val classes = cache.classes.toMutableSet()
                 cache.classProxy
                     .filter { it.proxyUsed }.forEach { proxy ->
-                        mutableClassList[proxy.originalIndex] = proxy.mutatedClass
+                        classes.remove(classes.elementAt(proxy.originalIndex))
+                        classes.add(proxy.mutatedClass)
                     }
-                return mutableClassList.toSet()
+                return classes
             }
 
             override fun getOpcodes(): Opcodes {