diff --git a/src/compiler.jl b/src/compiler.jl
index 12f0c47ab4..d45941708b 100644
--- a/src/compiler.jl
+++ b/src/compiler.jl
@@ -7142,8 +7142,8 @@ function _link(@nospecialize(job::CompilerJob{<:EnzymeTarget}), mod::LLVM.Module
     end
 
     # Now invoke the JIT
-    jitted_mod = JIT.add!(mod)
-    adjoint_addr = JIT.lookup(adjoint_name)
+    jit_dylib = JIT.add!(mod)
+    adjoint_addr = JIT.lookup(jit_dylib, adjoint_name)
 
     adjoint_ptr = pointer(adjoint_addr)
     if adjoint_ptr === C_NULL
@@ -7157,7 +7157,7 @@ function _link(@nospecialize(job::CompilerJob{<:EnzymeTarget}), mod::LLVM.Module
     if primal_name isa Nothing
         primal_ptr = C_NULL
     else
-        primal_addr = JIT.lookup(primal_name)
+        primal_addr = JIT.lookup(jit_dylib, primal_name)
         primal_ptr = pointer(primal_addr)
         if primal_ptr === C_NULL
             throw(
diff --git a/src/compiler/orcv2.jl b/src/compiler/orcv2.jl
index 79f076b2c4..1457edf56c 100644
--- a/src/compiler/orcv2.jl
+++ b/src/compiler/orcv2.jl
@@ -302,7 +302,7 @@ function add!(mod)
     jd = LLVM.JITDylib(lljit)
     tsm = move_to_threadsafe(mod)
     LLVM.add!(lljit, jd, tsm)
-    return nothing
+    return jd
 end
 
 function lookup(name)
@@ -310,4 +310,8 @@ function lookup(name)
     LLVM.lookup(lljit, JITDylib(lljit), name)
 end
 
+function lookup(jd::JITDylib, name)
+    LLVM.lookup(jit[].jit, jd, name)
+end
+
 end # module