Correct rank annotation for unlifted jitted scalar kernels