improved JITFunction correctness