Fix ML service test