Add unit correction to fix runaway update cycle (OMS-558)