Fix savestates under JIT