Fix bugs and typos (OMS-396)