Modify test cases for Travis CI again