Improve MachineLearningWebServiceTest and fix MainTest