Minor tweaks (OMS-345)