diff --git a/src/main/kotlin/app/revanced/patcher/extensions/Extensions.kt b/src/main/kotlin/app/revanced/patcher/extensions/Extensions.kt
index 81bb559..73d809b 100644
--- a/src/main/kotlin/app/revanced/patcher/extensions/Extensions.kt
+++ b/src/main/kotlin/app/revanced/patcher/extensions/Extensions.kt
@@ -3,6 +3,7 @@ package app.revanced.patcher.extensions
 import org.jf.dexlib2.AccessFlags
 import org.jf.dexlib2.builder.BuilderInstruction
 import org.jf.dexlib2.builder.MutableMethodImplementation
+import org.jf.dexlib2.iface.ClassDef
 
 infix fun AccessFlags.or(other: AccessFlags) = this.value or other.value
 
@@ -10,4 +11,9 @@ fun MutableMethodImplementation.addInstructions(index: Int, instructions: List<B
     for (i in instructions.lastIndex downTo 0) {
         this.addInstruction(index, instructions[i])
     }
-}
\ No newline at end of file
+}
+
+internal fun MutableSet<ClassDef>.replace(originalIndex: Int, mutatedClass: ClassDef) {
+    this.remove(this.elementAt(originalIndex))
+    this.add(mutatedClass)
+}