Add missing arguments (OMS-577)