Add hook for JIT compiler