Fix the test in travis