- Re-implemented return in JIT