Optimize the state machine update JIT