Code review fixes (OMS-609)