Fix jit.off invocations

Forgot the first, very relevant, parameter.
Spotted by @shalabhc fortunately.