Fix JIT backend after memoize_method change