diff --git a/src/enzyme_ad/jax/cpu.cc b/src/enzyme_ad/jax/cpu.cc
index 1b13d14c2..9e1e70c1e 100644
--- a/src/enzyme_ad/jax/cpu.cc
+++ b/src/enzyme_ad/jax/cpu.cc
@@ -11,5 +11,5 @@ void forwarding_custom_call(void *out, const void **in, const CallInfo *opaque,
 
 extern "C" void RegisterEnzymeXLACPUHandler() {
   CustomCallTargetRegistry::Global()->Register("enzymexla_compile_cpu",
-                                               forwarding_custom_call, "Host");
+                                               &forwarding_custom_call, "Host");
 }