Fix state update (OMS-198)
Merge pull request #22 from opennode/oms-198

Fix state update (OMS-198)