diff --git a/ext/ReactantKernelAbstractionsExt.jl b/ext/ReactantKernelAbstractionsExt.jl
index ee13c3bb3f..1e38b81b9a 100644
--- a/ext/ReactantKernelAbstractionsExt.jl
+++ b/ext/ReactantKernelAbstractionsExt.jl
@@ -109,12 +109,11 @@ function (obj::KA.Kernel{ReactantBackend})(args...; ndrange=nothing, workgroupsi
     return nothing
 end
 
-Reactant.@reactant_overlay @noinline Base.@nospecializeinfer function (
+Reactant.@reactant_overlay Base.@nospecializeinfer @noinline function (
     obj::KA.Kernel{ReactantBackend}
 )(
-    args...; ndrange=nothing, workgroupsize=nothing
+    @nospecialize args...; ndrange=nothing, workgroupsize=nothing
 )
-    @nospecialize
     return Reactant.call_with_reactant(
         Reactant.ka_with_reactant, ndrange, workgroupsize, obj, args...
     )
diff --git a/src/utils.jl b/src/utils.jl
index c7cb254946..eb8e2b3701 100644
--- a/src/utils.jl
+++ b/src/utils.jl
@@ -610,7 +610,10 @@ end
 #      using a custom interpreter in type unstable code.
 # `redub_arguments` is `(typeof(original_function), map(typeof, original_args_tuple)...)`
 function call_with_reactant_generator(
-    world::UInt, source::LineNumberNode, self, @nospecialize(redub_arguments)
+    world::UInt,
+    source::Union{LineNumberNode,Core.Method},
+    self,
+    @nospecialize(redub_arguments)
 )
     @nospecialize
     args = redub_arguments