Fix lambda arguments (OMS-194)