Fix the test script on Travis CI