Test fixes (OMS-287)