Fix the tests in travis