Modify test cases for Travis CI