Fix jit classes tests use of compile_extra