diff --git a/patches/src/main/kotlin/app/revanced/patches/shared/extension/SharedExtensionPatch.kt b/patches/src/main/kotlin/app/revanced/patches/shared/extension/SharedExtensionPatch.kt
index 386819fb5..40a0caf1b 100644
--- a/patches/src/main/kotlin/app/revanced/patches/shared/extension/SharedExtensionPatch.kt
+++ b/patches/src/main/kotlin/app/revanced/patches/shared/extension/SharedExtensionPatch.kt
@@ -33,6 +33,7 @@ class ExtensionHook internal constructor(
     private val insertIndexResolver: ((Method) -> Int),
     private val contextRegisterResolver: (Method) -> String,
 ) {
+    context(BytecodePatchContext)
     operator fun invoke(extensionClassDescriptor: String) {
         val insertIndex = insertIndexResolver(fingerprint.method)
         val contextRegister = contextRegisterResolver(fingerprint.method)