Fix state update (OMS-198)